diff --git a/tests/units/models/test_model_base.py b/tests/units/models/test_model_base.py
index 8a89d49a..7b3ff9e0 100644
--- a/tests/units/models/test_model_base.py
+++ b/tests/units/models/test_model_base.py
@@ -17,9 +17,9 @@ def __init__(self):
 		learn_return = fake_model.learn("current_state",
 			"action", "reward", "next_state")
 		predict_return = fake_model.predict("state")
-		learning_dict = fake_model.learning_dict
+		learning_data = fake_model.learning_data
 
 		assert isinstance(ModelBase, ABCMeta)
 		assert learn_return is None
 		assert predict_return is None
-		assert learning_dict is None
+		self.assertEqual(learning_data, {})
diff --git a/tests/units/sc2/actions/test_sc2_action.py b/tests/units/sc2/actions/test_sc2_action.py
index cd4a8e09..865b3a94 100644
--- a/tests/units/sc2/actions/test_sc2_action.py
+++ b/tests/units/sc2/actions/test_sc2_action.py
@@ -2,7 +2,7 @@
 
 from pysc2.lib import actions
 
-from urnai.actions.sc2_action import SC2Action
+from urnai.sc2.actions.sc2_action import SC2Action
 
 _BUILD_REFINERY = actions.RAW_FUNCTIONS.Build_Refinery_pt
 _NO_OP = actions.FUNCTIONS.no_op