Fix test to work with new db interface