Add two missing lines