Fix mistake from previous commit