diff --git a/tst/scripts/binary/binary.py b/tst/scripts/binary/binary.py
index ce1173c..bc6451f 100644
--- a/tst/scripts/binary/binary.py
+++ b/tst/scripts/binary/binary.py
@@ -32,8 +32,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "binary"
diff --git a/tst/scripts/binary_adi/binary_adi.py b/tst/scripts/binary_adi/binary_adi.py
index 63dcc66..5907c69 100644
--- a/tst/scripts/binary_adi/binary_adi.py
+++ b/tst/scripts/binary_adi/binary_adi.py
@@ -31,8 +31,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "binary_adi"
diff --git a/tst/scripts/collisions/collisions.py b/tst/scripts/collisions/collisions.py
index 1df597f..f9ce5c7 100644
--- a/tst/scripts/collisions/collisions.py
+++ b/tst/scripts/collisions/collisions.py
@@ -28,8 +28,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "collisions"
diff --git a/tst/scripts/coords/blast.py b/tst/scripts/coords/blast.py
index 872fa56..03a9404 100644
--- a/tst/scripts/coords/blast.py
+++ b/tst/scripts/coords/blast.py
@@ -27,8 +27,7 @@
 import h5py
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "blast"
diff --git a/tst/scripts/diffusion/alpha_disk.py b/tst/scripts/diffusion/alpha_disk.py
index faf2495..2a28ba4 100644
--- a/tst/scripts/diffusion/alpha_disk.py
+++ b/tst/scripts/diffusion/alpha_disk.py
@@ -28,8 +28,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "alpha_disk"
diff --git a/tst/scripts/diffusion/thermal_diffusion.py b/tst/scripts/diffusion/thermal_diffusion.py
index 26d772d..a76ce98 100644
--- a/tst/scripts/diffusion/thermal_diffusion.py
+++ b/tst/scripts/diffusion/thermal_diffusion.py
@@ -28,8 +28,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "thermal_diffusion"
diff --git a/tst/scripts/diffusion/viscous_diffusion.py b/tst/scripts/diffusion/viscous_diffusion.py
index 19f1d9a..c3e44d8 100644
--- a/tst/scripts/diffusion/viscous_diffusion.py
+++ b/tst/scripts/diffusion/viscous_diffusion.py
@@ -27,8 +27,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "viscous_diffusion"
diff --git a/tst/scripts/disk/disk.py b/tst/scripts/disk/disk.py
index 7a1fe83..d57edb7 100644
--- a/tst/scripts/disk/disk.py
+++ b/tst/scripts/disk/disk.py
@@ -29,8 +29,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "disk"
diff --git a/tst/scripts/disk_nbody/disk_nbody.py b/tst/scripts/disk_nbody/disk_nbody.py
index 1f69e80..915c273 100644
--- a/tst/scripts/disk_nbody/disk_nbody.py
+++ b/tst/scripts/disk_nbody/disk_nbody.py
@@ -29,8 +29,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "disk_nbody"
diff --git a/tst/scripts/drag/drag.py b/tst/scripts/drag/drag.py
index fb42312..b61a674 100644
--- a/tst/scripts/drag/drag.py
+++ b/tst/scripts/drag/drag.py
@@ -28,8 +28,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "drag"
diff --git a/tst/scripts/nbody/nbody.py b/tst/scripts/nbody/nbody.py
index a98b5ea..9b92c59 100644
--- a/tst/scripts/nbody/nbody.py
+++ b/tst/scripts/nbody/nbody.py
@@ -33,8 +33,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "nbody"
diff --git a/tst/scripts/ssheet/ssheet.py b/tst/scripts/ssheet/ssheet.py
index 4e6c65a..00f75dc 100644
--- a/tst/scripts/ssheet/ssheet.py
+++ b/tst/scripts/ssheet/ssheet.py
@@ -30,8 +30,7 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import matplotlib
-if not os.getenv('DISPLAY'):
-    matplotlib.use('Agg')  # Use the Agg backend if no display is available
+matplotlib.use('Agg')  # Use the Agg backend to avoid issues with DISPLAY not being set
 
 _nranks = 1
 _file_id = "ssheet"