Fix code block to be consistent with the other