Forgot two lines in previous commit