From ce6ca64b4d15eb32b27a73899029637e292a335b Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Fri, 7 Oct 2022 12:00:04 +0530
Subject: [PATCH] Agent: Define exploiter and propagation tags correctly

---
 monkey/infection_monkey/exploit/smbexec.py | 28 ++++++++++------------
 1 file changed, 12 insertions(+), 16 deletions(-)

diff --git a/monkey/infection_monkey/exploit/smbexec.py b/monkey/infection_monkey/exploit/smbexec.py
index fdcd6ee00..16ea4d0e1 100644
--- a/monkey/infection_monkey/exploit/smbexec.py
+++ b/monkey/infection_monkey/exploit/smbexec.py
@@ -49,22 +49,18 @@ class SMBExploiter(HostExploiter):
     }
     USE_KERBEROS = False
     SMB_SERVICE_NAME = "InfectionMonkey"
-
-    def _EXPLOITER_TAGS(self) -> Tuple[str, ...]:
-        return (
-            SMBEXEC_EXPLOITER_TAG,
-            T1021_ATTACK_TECHNIQUE_TAG,
-            T1110_ATTACK_TECHNIQUE_TAG,
-            T1210_ATTACK_TECHNIQUE_TAG,
-        )
-
-    def _PROPAGATION_TAGS(self) -> Tuple[str, ...]:
-        return (
-            SMBEXEC_EXPLOITER_TAG,
-            T1021_ATTACK_TECHNIQUE_TAG,
-            T1105_ATTACK_TECHNIQUE_TAG,
-            T1210_ATTACK_TECHNIQUE_TAG,
-        )
+    _EXPLOITER_TAGS = (
+        SMBEXEC_EXPLOITER_TAG,
+        T1021_ATTACK_TECHNIQUE_TAG,
+        T1110_ATTACK_TECHNIQUE_TAG,
+        T1210_ATTACK_TECHNIQUE_TAG,
+    )
+    _PROPAGATION_TAGS = (
+        SMBEXEC_EXPLOITER_TAG,
+        T1021_ATTACK_TECHNIQUE_TAG,
+        T1105_ATTACK_TECHNIQUE_TAG,
+        T1210_ATTACK_TECHNIQUE_TAG,
+    )
 
     def _exploit_host(self):