Fix incorrect model reference