fix BaseModel