Fix regression I caused :(