diff --git a/.vscode/settings.json b/.vscode/settings.json
index b1985ef..170e4c3 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -10,5 +10,19 @@
         "Compile current GSC File": [
             "*.gsc"
         ]
-    }
+    },
+    "Lua.diagnostics.globals": [
+        "InheritFrom",
+        "CoD",
+        "LUI",
+        "Dvar",
+        "Engine",
+        "UIExpression",
+        "RegisterMaterial",
+        "GetTextDimensions",
+        "setstencil",
+        "HUD_IngameMenuClosed",
+        "HUD_IsFFA",
+        "HideGlobe"
+    ]
 }
\ No newline at end of file
diff --git a/raw/scripts/mp/max_allocation_fix_source-compiled.gsc b/raw/scripts/mp/max_allocation_fix_source-compiled.gsc
index 9a69d07..834441b 100644
Binary files a/raw/scripts/mp/max_allocation_fix_source-compiled.gsc and b/raw/scripts/mp/max_allocation_fix_source-compiled.gsc differ
diff --git a/raw/scripts/mp/max_allocation_fix_source.gsc b/raw/scripts/mp/max_allocation_fix_source.gsc
index 96e4d03..7afddde 100644
--- a/raw/scripts/mp/max_allocation_fix_source.gsc
+++ b/raw/scripts/mp/max_allocation_fix_source.gsc
@@ -16,6 +16,8 @@ on_player_connect()
 
 check_player_classes()
 {
+	self endon( "disconnect" );
+
 	for ( class_num = 0; class_num < 10; class_num++ )
 	{
 		allocationSpent = self GetLoadoutAllocation( class_num );
@@ -27,7 +29,7 @@ check_player_classes()
 			logline1 = "Player: " + self.name + " XUID: " + self getXUID() + " had too many items in their class.";
 			print( logline1 );
 			kick( self getEntityNumber() );
-			break;
+			return;
 		}
 	}
 }
\ No newline at end of file