Fix up mocked database