Fix user model code