A minor fix

Signed-off-by: Shwetank Dixit <emailshwetank@gmail.com>