diff --git a/modules/exploits/windows/misc/delta_electronics_infrasuite_deserialization.rb b/modules/exploits/windows/misc/delta_electronics_infrasuite_deserialization.rb
index 99d7e9ecd2..24ef46144e 100644
--- a/modules/exploits/windows/misc/delta_electronics_infrasuite_deserialization.rb
+++ b/modules/exploits/windows/misc/delta_electronics_infrasuite_deserialization.rb
@@ -130,4 +130,8 @@ class MetasploitModule < Msf::Exploit::Remote
     pkt = "\x01#{[ serialized.length ].pack('n')}#{serialized}"
     udp_sock.put(pkt)
   end
+
+  def cleanup
+    disconnect_udp
+  end
 end