Fixed exit of main and pygame gui

overcooked_simulator/__main__.py

@@ -44,7 +44,7 @@ def main(cli_args=None):
             game_server.terminate()
         if pygame_gui is not None and pygame_gui.is_alive():
             print("Terminate pygame gui")
-            game_server.terminate()
+            pygame_gui.terminate()
         time.sleep(0.1)
 
 

overcooked_simulator/gui_2d_vis/overcooked_gui.py

@@ -669,6 +669,7 @@ class PyGameGUI:
             except (KeyboardInterrupt, SystemExit):
                 self.running = False
 
+        self.disconnect_websockets()
         pygame.quit()
         sys.exit()