Fix compilation warning of last commit

Signed-off-by: Sup Yut Sum <ch3cooli@gmail.com>