diff --git a/zulip_bots/zulip_bots/bots/connect_four/test_connect_four.py b/zulip_bots/zulip_bots/bots/connect_four/test_connect_four.py
index c1d15ba..792658c 100644
--- a/zulip_bots/zulip_bots/bots/connect_four/test_connect_four.py
+++ b/zulip_bots/zulip_bots/bots/connect_four/test_connect_four.py
@@ -1,6 +1,6 @@
 from zulip_bots.test_lib import BotTestCase, DefaultTests
 
-from zulip_bots.bots.connect_four.connect_four import *
+from zulip_bots.bots.connect_four.connect_four import ConnectFourModel
 from zulip_bots.game_handler import BadMoveException
 from typing import Dict, List