Tests with models are now working