Let's try just testing the return value
Since extending the model class breaks the other tests.