Fix test due to db changes