Fixed bug
Signed-off-by: abhipranay