fix small model error