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