Fix error from previous commit