fix the error caused by refactor