fixed model error in last commit