Fixed compilation errors

Signed-off-by: Arif <arif@bansberrysg.com>