diff --git a/pokemonterminal/load_all_pokemon.py b/tests/load_all_pokemon.py
similarity index 96%
rename from pokemonterminal/load_all_pokemon.py
rename to tests/load_all_pokemon.py
index 2a900bc..50fe28d 100644
--- a/pokemonterminal/load_all_pokemon.py
+++ b/tests/load_all_pokemon.py
@@ -6,7 +6,8 @@ import os
 from pokemonterminal.database import Pokemon
 
 
-SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
+SCRIPT_DIR = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
+SCRIPT_DIR = os.path.join(SCRIPT_DIR, "pokemonterminal")
 DATA_DIR = os.path.join(SCRIPT_DIR, 'Data')
 IMAGES_DIR = os.path.join(SCRIPT_DIR, 'Images')
 EXTRA_DIR = os.path.join(IMAGES_DIR, 'Extra')
diff --git a/tests/test_load.py b/tests/test_load.py
index e5b146d..10feb86 100644
--- a/tests/test_load.py
+++ b/tests/test_load.py
@@ -3,7 +3,7 @@
 # To run the tests, use: python3 -m pytest --capture=sys
 
 from pokemonterminal.database import Database, Pokemon
-from pokemonterminal.load_all_pokemon import load_all_pokemon
+from tests.load_all_pokemon import load_all_pokemon
 from tests.test_utils import expected_len, MAX_ID