refactor
fix compile error