Fix up db for testing