Fixed comment bug in previous commit

Signed-off-by: Archit Shukla <architshukla2@gmail.com>