Fixing the build failure issue;

Signed-off-by: sagar <expie00101@gmail.com>