diff --git a/baselines/common/tests/test_cartpole.py b/baselines/common/tests/test_cartpole.py
index 3c33b8f..359006c 100644
--- a/baselines/common/tests/test_cartpole.py
+++ b/baselines/common/tests/test_cartpole.py
@@ -19,7 +19,6 @@ learn_kwargs = {
     'trpo_mpi': {}
 }
 
-
 @pytest.mark.slow
 @pytest.mark.parametrize("alg", learn_kwargs.keys())
 def test_cartpole(alg):
@@ -39,7 +38,3 @@ def test_cartpole(alg):
         return env
 
     reward_per_episode_test(env_fn, learn_fn, 100)
-    
-
-if __name__ == '__main__':
-    test_cartpole('a2c')