fix refactoring error
fix model name