Fix test_model_creation failure