Added missed change to previous commit