Fix test code comment