fix comment for rearranged code