Fixed model function argument order bug