fix variable missed in refactoring