Update: fix compile error.