Fix build in linux

Signed-off-by: Balaji Rao <balaji@raobalaji.com>