diff --git a/tests/example_load_data.py b/tests/example_load_data.py
index a30e2f1..c630e0b 100644
--- a/tests/example_load_data.py
+++ b/tests/example_load_data.py
@@ -9,7 +9,7 @@
 from decimal import Decimal
 import geopandas as gpd
 
-from hydrocronapi.controllers.db.hydrocron_database import Hydrocron_DB
+from ..hydrocronapi.controllers.db.hydrocron_database import Hydrocron_DB
 
 
 # ## Set up the connection to dynamoDB