Make test database agnostic