Fix build

Signed-off-by: Hieu Phan <aznrice2k4@gmail.com>