diff --git a/wrolpi/db.py b/wrolpi/db.py
index 8958f677..4e8a7572 100644
--- a/wrolpi/db.py
+++ b/wrolpi/db.py
@@ -72,7 +72,7 @@ def get_db_context() -> Tuple[create_engine, Session]:
     Get a DB engine and session.
     """
     local_engine, session = _get_db_session()
-    if PYTEST and 'wrolpi_testing' not in str(local_engine.url):
+    if PYTEST and not local_engine.url.database.startswith('wrolpi_testing_'):
         raise ValueError(f'Running tests, but a test database is not being used!! {local_engine.url=}')
     return local_engine, session