removed unnecessary comments and fixed the failing test case

Signed-off-by: Ayudh Das <ayudh.d@gmail.com>