diff --git a/TestON/core/utilities.py b/TestON/core/utilities.py
index be15bc1..ed37ae3 100644
--- a/TestON/core/utilities.py
+++ b/TestON/core/utilities.py
@@ -336,12 +336,10 @@
             return ret
         except AssertionError:
             main.log.exception( "Invalid arguements for retry: " )
-            main.cleanup()
-            main.exit()
+            main.cleanAndExit()
         except Exception:
             main.log.exception( "Uncaught exception in retry: " )
-            main.cleanup()
-            main.exit()
+            main.cleanAndExit()
 
 
 if __name__ != "__main__":