Typo
Fixed test after changing model