Add back comment that was accidentally removed (minor)