Bug fix

Signed-off-by: Gaurav Munjal <Gaurav0@aol.com>