small followup changes