add 2 missing lines