Update test code comment