Updated test code to match new model creation method signature.