From a4092acf5b313ace94a6910c9a15252e9c42b2be Mon Sep 17 00:00:00 2001
From: Pieter-Jan Briers <pieterjan.briers+git@gmail.com>
Date: Mon, 13 Apr 2020 22:27:47 +0200
Subject: [PATCH] Fix not returning to main menu on disconnect correctly.

---
 Content.Client/EntryPoint.cs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/Content.Client/EntryPoint.cs b/Content.Client/EntryPoint.cs
index 1cc3c720a0..ff63ab58ad 100644
--- a/Content.Client/EntryPoint.cs
+++ b/Content.Client/EntryPoint.cs
@@ -15,6 +15,7 @@ using Content.Shared.GameObjects.Components.Chemistry;
 using Content.Shared.GameObjects.Components.Markers;
 using Content.Shared.GameObjects.Components.Research;
 using Content.Shared.GameObjects.Components.VendingMachines;
+using Robust.Client;
 using Robust.Client.Interfaces;
 using Robust.Client.Interfaces.Graphics.Overlays;
 using Robust.Client.Interfaces.Input;
@@ -227,6 +228,14 @@ namespace Content.Client
             IoCManager.Resolve<IClientPreferencesManager>().Initialize();
             IoCManager.Resolve<IItemSlotManager>().Initialize();
 
+            _baseClient.RunLevelChanged += (sender, args) =>
+            {
+                if (args.NewLevel == ClientRunLevel.Initialize)
+                {
+                    _stateManager.RequestStateChange<MainScreen>();
+                }
+            };
+
             // Fire off into state dependent on launcher or not.
             if (_gameController.LaunchState.FromLauncher)
             {