fixed missing change in the previous commit