diff --git a/test/test_ppo.py b/test/test_ppo.py
index f4a965ce2..9ddaab8be 100644
--- a/test/test_ppo.py
+++ b/test/test_ppo.py
@@ -6,7 +6,7 @@
 import gym
 import tensorflow as tf
 
-from spinup import ppo
+from spinup import ppo_tf1 as ppo
 
 
 class TestPPO(unittest.TestCase):