Fixing issue with initial database connection when running test database