fixed missing fix from previous commit