Fixes Build error

Signed-off-by: hegdeashwin <ashwin.hegde3@gmail.com>