fix refactoring error

#323