Fix the test defining the models in the right place