Fix broken imports in db_tests.py