diff --git a/ldv_migrate/ldv_migrate/connection/retrieve_db_connection.py b/ldv_migrate/ldv_migrate/connection/retrieve_db_connection.py
index 251cfb68520dad125b8fc138af585b75ec11fedc..5fbb505c88830cabff072ef65a7fa54aa352fa4e 100644
--- a/ldv_migrate/ldv_migrate/connection/retrieve_db_connection.py
+++ b/ldv_migrate/ldv_migrate/connection/retrieve_db_connection.py
@@ -1,7 +1,11 @@
 import sys
 
 import psycopg2
-from connection.config import read_config
+try:
+    from connection.config import read_config
+except:
+    from ldv_migrate.connection.config import read_config
+
 import logging
 import argparse
 from sshtunnel import SSHTunnelForwarder