diff --git a/makefile_and_run.py b/makefile_and_run.py
index 71bedbe..999f9c1 100755
--- a/makefile_and_run.py
+++ b/makefile_and_run.py
@@ -160,6 +160,8 @@ def _gpu_runtime_env():
         "AMSS_CUDA_DEVICE_SEGMENT_BATCH": "0",
         "AMSS_CUDA_UNCACHED_DEVICE_BUFFERS": "1",
     }
+    if getattr(input_data, "Equation_Class", "") == "BSSN":
+        defaults["AMSS_CUDA_AMR_RESTRICT_DEVICE"] = "1"
     if getattr(input_data, "Equation_Class", "") == "Z4C":
         defaults.update({
             "AMSS_Z4C_CUDA_RESIDENT": "1",