fix compile error, clean code.