Correct proof of lemma 3.16

I'd commit this normally, but I'm unsure of the maths. I think what I changed makes more sense, but someone should have a look at it.