diff --git a/connectx/connectx_gym/connectx_env.py b/connectx/connectx_gym/connectx_env.py
index cac0572..526f2ef 100644
--- a/connectx/connectx_gym/connectx_env.py
+++ b/connectx/connectx_gym/connectx_env.py
@@ -34,7 +34,6 @@ def __init__(
 
         self.action_space = act_space
         self.obs_space = obs_space
-        self.default_reward_space = GameResultReward()
         self.info = dict()
 
     def reset(self, **kwargs):
diff --git a/connectx/connectx_gym/reward_spaces.py b/connectx/connectx_gym/reward_spaces.py
index 84ebd1f..0c47a64 100644
--- a/connectx/connectx_gym/reward_spaces.py
+++ b/connectx/connectx_gym/reward_spaces.py
@@ -6,7 +6,7 @@
 from kaggle_environments.core import Environment
 import numpy as np
 
-from connectx_env import ConnectFour
+from .connectx_env import ConnectFour
 from ..utility_constants import BOARD_SIZE
 
 class RewardSpec(NamedTuple):