diff --git a/dreamer.py b/dreamer.py
index 3b98259..61ae538 100644
--- a/dreamer.py
+++ b/dreamer.py
@@ -213,7 +213,7 @@ def make_env(config, logger, mode, train_eps, eval_eps):
     elif suite == "mazegym":
         import gym
         env = gym.make('memory_maze:MemoryMaze-9x9-v0')
-        from envs.memmaze import MZGymWrapper
+        from envs.memmazeEnv import MZGymWrapper
         env = MZGymWrapper(env)
         
         env = wrappers.OneHotAction(env)
diff --git a/envs/memmaze.py b/envs/memmazeEnv.py
similarity index 99%
rename from envs/memmaze.py
rename to envs/memmazeEnv.py
index 9de34c4..baabf90 100644
--- a/envs/memmaze.py
+++ b/envs/memmazeEnv.py
@@ -9,7 +9,7 @@ import gym
 import numpy as np
 
 
-class GymWrapper:
+class MZGymWrapper:
 
   def __init__(self, env, obs_key='image', act_key='action'):
     self._env = env