Fix to add test to db