fixed bug in prepare_model_for_test when classes that inherits from model.Model were passed as parameters to the function