Fix error in previous commit