Attempt to fix compile error.

Signed-off-by: Max Fu <fxy83613@gmail.com>