diff --git a/.gitignore b/.gitignore
index 579da0c..cb1d30e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
+logs
 novgrid/baselines/logs/*
 novgrid/baselines/logs
 *.pyc
diff --git a/novgrid/envs/lavagapdoorkey.py b/novgrid/envs/lavagapdoorkey.py
index fbc0342..fde2e9c 100644
--- a/novgrid/envs/lavagapdoorkey.py
+++ b/novgrid/envs/lavagapdoorkey.py
@@ -193,7 +193,7 @@ def step(self, action, **kwargs):
         obs, reward, done, info = super().step(action)
         if done and fwd_cell and fwd_cell.type == 'lava':
             self.agent_pos = fwd_pos
-            obs = self.gen_obs()['image']
+            obs = self.gen_obs()
             done = False
         return obs, reward, done, info