Changes to make it work

Signed-off-by: Rohit Jain <rohit.jain@one97.net>