修正 model类