Working on the model integration test