Fix broken tests
Fix a broken database test