fix two lines missed in previous commit