Updated test after changes in model