Fix code comment with last commit