diff --git a/monkey/infection_monkey/exploit/tools/wmi_tools.py b/monkey/infection_monkey/exploit/tools/wmi_tools.py
index 078d37daa..b7eaf8675 100644
--- a/monkey/infection_monkey/exploit/tools/wmi_tools.py
+++ b/monkey/infection_monkey/exploit/tools/wmi_tools.py
@@ -88,7 +88,7 @@ class WmiTools(object):
         for port_map in list(DCOMConnection.PORTMAPS.keys()):
             del DCOMConnection.PORTMAPS[port_map]
         for oid_set in list(DCOMConnection.OID_SET.keys()):
-            del DCOMConnection.OID_SET[port_map]
+            del DCOMConnection.OID_SET[oid_set]
 
         DCOMConnection.OID_SET = {}
         DCOMConnection.PORTMAPS = {}