diff --git a/columbus/env.py b/columbus/env.py
index edb70e2..aff4228 100644
--- a/columbus/env.py
+++ b/columbus/env.py
@@ -95,6 +95,8 @@ class ColumbusEnv(gym.Env):
 
         self._init = False
 
+        self.is_columbus_env = True
+
     @property
     def observation_space(self):
         if not self._init: