fix model() function