Added back two missing lines