fix code review
Remove print statement
Fix the test name