Fix incorrect model size.