diff --git a/python/gtsam/tests/test_DiscreteFactorGraph.py b/python/gtsam/tests/test_DiscreteFactorGraph.py
index afc6630bd..9ed7cc010 100644
--- a/python/gtsam/tests/test_DiscreteFactorGraph.py
+++ b/python/gtsam/tests/test_DiscreteFactorGraph.py
@@ -336,10 +336,6 @@ class TestDiscreteFactorGraph(GtsamTestCase):
 # }
 # #endif
 
-# /* ************************************************************************* */
-# int main() {
-# TestResult tr
-# return TestRegistry::runAllTests(tr)
-# }
-# /* ************************************************************************* */
 
+if __name__ == "__main__":
+    unittest.main()