fix the multiple model bug