Fixing RegisterModel