Fix incorrect model naming