diff --git a/ai-moderation/app/container.py b/ai-moderation/app/container.py
deleted file mode 100644
index ef089ff..0000000
--- a/ai-moderation/app/container.py
+++ /dev/null
@@ -1,14 +0,0 @@
-from app.moderation.profanity.detector import ProfanityDetector
-from app.moderation.profanity.dictionary import ProfanityDictionary
-from app.moderation.profanity.lemmatizer import Lemmatizer
-
-dictionary = ProfanityDictionary(
-    "app/resources/profanity_words.txt"
-)
-
-lemmatizer = Lemmatizer()
-
-profanity_detector = ProfanityDetector(
-    dictionary,
-    lemmatizer
-)
diff --git a/ai-moderation/app/moderation/profanity/detector.py b/ai-moderation/app/moderation/profanity/detector.py
index abbfc56..e2b1e6e 100644
--- a/ai-moderation/app/moderation/profanity/detector.py
+++ b/ai-moderation/app/moderation/profanity/detector.py
@@ -33,7 +33,7 @@ class ProfanityDetector:
 
       for bad_word in self.dictionary.words:
 
-        if lemma.startswith(bad_word):
+        if lemma == bad_word or lemma.startswith(bad_word + "-"):
 
           result.append(word)