Fixes build error.

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