diff --git a/cscs-checks/mch/cuda_stress_test.py b/cscs-checks/mch/cuda_stress_test.py
index eb363349cf..76eff4ae64 100644
--- a/cscs-checks/mch/cuda_stress_test.py
+++ b/cscs-checks/mch/cuda_stress_test.py
@@ -8,7 +8,7 @@
 
 
 @rfm.simple_test
-class CudaStressTest(rfm.RegressionTest):
+class cuda_stress_test(rfm.RegressionTest):
     descr = 'MCH CUDA stress test'
     valid_systems = ['daint:gpu', 'dom:gpu', 'arolla:cn', 'tsa:cn']
     valid_prog_environs = ['*']
@@ -25,9 +25,14 @@ def set_environment(self):
             self.exclusive_access = True
             self.valid_prog_environs = ['PrgEnv-gnu', 'PrgEnv-gnu-nompi',
                                         'PrgEnv-pgi', 'PrgEnv-pgi-nompi']
-            self.modules = ['cuda/10.1.243']
         else:
-            self.valid_prog_environs = ['PrgEnv-gnu']
+            self.valid_prog_environs = ['PrgEnv-gnu', 'PrgEnv-nvidia']
+
+    @run_after('setup')
+    def set_modules(self):
+        if self.current_system.name in {'arolla', 'tsa'}:
+            self.modules = ['cuda/10.1.243']
+        elif self.current_environ.name != 'PrgEnv-nvidia':
             self.modules = ['craype-accel-nvidia60', 'cdt-cuda']
 
     @run_before('compile')