diff --git a/model/project.py b/model/project.py
index bd9c377..89e1ea4 100644
--- a/model/project.py
+++ b/model/project.py
@@ -46,6 +46,15 @@ def prepare_project(project_name, settings):
             continue
         if file.startswith("."):
             continue
+        if file.endswith(".exe"):
+            # keep all exes except:
+            if file != "main.exe" and not file.endswith(".infected.exe"):
+                continue
+        if file.endswith(".dll"):
+            # keep all dlls except:
+            if not file.endswith(".infected.dll"):
+                continue
+
         os.remove(dst + file)
 
     # copy *.c *.h files from src directory to dst directory
diff --git a/supermega.py b/supermega.py
index 4bc0273..aa7246e 100644
--- a/supermega.py
+++ b/supermega.py
@@ -86,7 +86,7 @@ def main():
             logger.info("Could not find: {}".format(args.inject))
             return
         settings.inject_exe_in = args.inject
-        settings.inject_exe_out = args.inject.replace(".exe", ".infected.exe")
+        settings.inject_exe_out = args.inject.replace(".exe", ".infected.exe").replace(".dll", ".infected.dll")
 
     settings.prep_web("default")
     write_webproject("default", settings)