diff --git a/grai-server/app/connections/tests/test_tasks.py b/grai-server/app/connections/tests/test_tasks.py
index 3849cc2ee..53ebb1eb1 100644
--- a/grai-server/app/connections/tests/test_tasks.py
+++ b/grai-server/app/connections/tests/test_tasks.py
@@ -259,6 +259,8 @@ def test_run_update_server_postgres_wrong_password(self, test_workspace, test_po
         assert (
             run.metadata["message"]
             == 'connection to server at "localhost" (127.0.0.1), port 5432 failed: FATAL:  password authentication failed for user "grai"\n'
+            or run.metadata["message"]
+            == 'connection to server at "127.0.0.1", port 5432 failed: FATAL:  password authentication failed for user "grai"'
         )
 
     def test_run_update_server_postgres_no_database(self, test_workspace, test_postgres_connector, test_source):
@@ -287,7 +289,7 @@ def test_run_update_server_postgres_no_database(self, test_workspace, test_postg
             run.metadata["message"]
             == 'connection to server at "localhost" (127.0.0.1), port 5432 failed: FATAL:  database "wrong" does not exist\n'
             or run.metadata["message"]
-            == 'connection to server at "127.0.0.1", port 5432 failed: FATAL:  password authentication failed for user "grai"'
+            == 'connection to server at "127.0.0.1", port 5432 failed: FATAL:  database "wrong" does not exist'
         )
 
     def test_run_update_server_no_connector(self, test_workspace, test_connector, test_source):