Patch the getModel method