Updated tests after model change