Adapt tests to the new database