fix missing lines from previous commit