followup fix for breaking test