From 879b823d02e556226c5bf21fca2c2192ce7fd8bf Mon Sep 17 00:00:00 2001
From: Mustafa Enes Batur <mustafa.enes99@gmail.com>
Date: Mon, 20 Nov 2023 13:50:48 +0100
Subject: [PATCH] Stop mujoco viewer on close

---
 fancy_gym/envs/mujoco/air_hockey/air_hockey_env_wrapper.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fancy_gym/envs/mujoco/air_hockey/air_hockey_env_wrapper.py b/fancy_gym/envs/mujoco/air_hockey/air_hockey_env_wrapper.py
index cfed75d..0e8e31e 100644
--- a/fancy_gym/envs/mujoco/air_hockey/air_hockey_env_wrapper.py
+++ b/fancy_gym/envs/mujoco/air_hockey/air_hockey_env_wrapper.py
@@ -152,7 +152,7 @@ class AirHockeyEnv(Environment):
         return self
     
     def close(self):
-        return
+        self.base_env.stop()
 
 
 if __name__ == "__main__":