Fix db tests