diff --git a/main.py b/main.py
index e9f43e1..e252288 100755
--- a/main.py
+++ b/main.py
@@ -75,7 +75,7 @@ def all_to_translate(_dir):
 
                 _msgid = f'msgid "{_name[0][0]}"\n'
 
-                _entity = f'#: {_file.split("WPE2")[1]}:\n{_msgctxt}{_msgid}msgstr ""\n\n'
+                _entity = f'#: {_file.split("weapon-project-extended")[1]}:\n{_msgctxt}{_msgid}msgstr ""\n\n'
                 _str = _str + _entity
     with open('test.txt', 'w') as f:
         f.write(_str)