diff --git a/scripts/zm/replaced/zm_tomb_dig.gsc b/scripts/zm/replaced/zm_tomb_dig.gsc
new file mode 100644
index 00000000..97e40896
--- /dev/null
+++ b/scripts/zm/replaced/zm_tomb_dig.gsc
@@ -0,0 +1,10 @@
+#include maps\mp\_utility;
+#include common_scripts\utility;
+#include maps\mp\zombies\_zm_utility;
+
+dig_disconnect_watch( n_player, v_origin, v_angles )
+{
+	self waittill( "disconnect" );
+	level setclientfield( "shovel_player" + n_player, 0 );
+	level setclientfield( "helmet_player" + n_player, 0 );
+}
\ No newline at end of file
diff --git a/scripts/zm/zm_tomb/zm_tomb_reimagined.gsc b/scripts/zm/zm_tomb/zm_tomb_reimagined.gsc
index e957d3f3..a1cceccb 100644
--- a/scripts/zm/zm_tomb/zm_tomb_reimagined.gsc
+++ b/scripts/zm/zm_tomb/zm_tomb_reimagined.gsc
@@ -3,10 +3,12 @@
 #include maps\mp\zombies\_zm_utility;
 
 #include scripts/zm/replaced/zm_tomb_craftables;
+#include scripts/zm/replaced/zm_tomb_dig;
 
 main()
 {
 	replaceFunc(maps/mp/zm_tomb_craftables::quadrotor_control_thread, scripts/zm/replaced/zm_tomb_craftables::quadrotor_control_thread);
+	replaceFunc(maps/mp/zm_tomb_dig::dig_disconnect_watch, scripts/zm/replaced/zm_tomb_dig::dig_disconnect_watch);
 }
 
 init()