diff --git a/src/moller/main.py b/src/moller/main.py
index 0d57bea..59fdcad 100644
--- a/src/moller/main.py
+++ b/src/moller/main.py
@@ -85,7 +85,7 @@ def generate(self, fp):
         lines = self.code.splitlines()
         lines_new = []
         for line in lines:
-            if re.match(r'(srun|mpirun|mpiexec)', line):
+            if re.search(r'\b(srun|mpirun|mpiexec)\b', line):
                 line = re.sub(r'(srun|mpirun|mpiexec)', srun_str, line)
 
                 lines_new.append(r'  DEBUG "$_work_item: ' + line + '"')