fix minor issue in BaseModel