made the tests pass again, only when other models are added we'll try to fix this issue