Fix mistake with previous commit