diff --git a/tools/payload_creator.py b/tools/payload_creator.py
index 05e1734..58029a3 100644
--- a/tools/payload_creator.py
+++ b/tools/payload_creator.py
@@ -1,4 +1,5 @@
 import os
+import subprocess
 
 from core import HackingTool, HackingToolsCollection, console
 
diff --git a/tools/wordlist_generator.py b/tools/wordlist_generator.py
index 677efac..44f9aab 100644
--- a/tools/wordlist_generator.py
+++ b/tools/wordlist_generator.py
@@ -3,6 +3,7 @@ import subprocess
 
 from rich.panel import Panel
 from rich.prompt import Prompt
+from rich.table import Table
 from rich import box
 
 from core import HackingTool, HackingToolsCollection, console