Fix code comment