Bug fix

Signed-off-by: Gaurav Munjal <gaurav9576@yahoo.com>