diff --git a/deism/tests/test_deism_arg.py b/deism/tests/test_deism_arg.py
index 15e8b71..2fe6fc1 100644
--- a/deism/tests/test_deism_arg.py
+++ b/deism/tests/test_deism_arg.py
@@ -134,6 +134,10 @@ def main():
         print("DEISM Function has not been run because --run flag was not set.\n")
 
 
+def test_main():
+    main()
+
+
 # -------------------------------------------------------
 if __name__ == "__main__":
-    main()
+    test_main()