Small comments fix