Update as per suggestion by cpg

Signed-off-by: Arpit Goyal <agmps17@gmail.com>