Fix tests that failed due to database update