diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index f51e7cb..0b5afa9 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -52,7 +52,7 @@ jobs:
           wget https://downloads.openmicroscopy.org/images/omero_db_searchengine.zip -P app_data
           unzip app_data/omero_db_searchengine.zip -d app_data/          
           # run  restore omero database
-          python manage.py restore_postgresql_database
+          python manage.py restore_postgresql_database -s omero1
           # run indexing indexing
           python manage.py get_index_data_from_database -b False -s omero1
           # run tests
diff --git a/omero_search_engine/database/utils.py b/omero_search_engine/database/utils.py
index 7186aa3..a5e065b 100644
--- a/omero_search_engine/database/utils.py
+++ b/omero_search_engine/database/utils.py
@@ -52,8 +52,8 @@ def restore_database(source):
             proc = subprocess.Popen(
                 restore_command,
                 shell=True,
-                env={"PGPASSWORD": search_omero_app.config.get("DATABASE_PASSWORD")},
+                env={"PGPASSWORD": data_source.get("DATABASE").get("DATABASE_PASSWORD")},
             )
             proc.wait()
         except Exception as e:
-            print("Error, exception happened during dump %s" % (e))
+            print("Error: exception happened during dump %s" % (e))