diff --git a/requirements.txt b/requirements.txt
index 15545ca7..693ec0c2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 torch-ac>=1.4.0
-minigrid
+minigrid>=2.2.0
 tensorboardX>=1.6
 numpy>=1.3
 gymnasium>=0.26
diff --git a/scripts/visualize.py b/scripts/visualize.py
index bec8430e..4dd2588c 100644
--- a/scripts/visualize.py
+++ b/scripts/visualize.py
@@ -76,12 +76,9 @@
         done = terminated | truncated
         agent.analyze_feedback(reward, done)
 
-        if done or env.window.closed:
+        if done:
             break
 
-    if env.window.closed:
-        break
-
 if args.gif:
     print("Saving gif... ", end="")
     write_gif(numpy.array(frames), args.gif+".gif", fps=1/args.pause)