The whole series relied on David not knowing what's going on. Letting the character relax / normalise at the end would probably block off a lot more plot choices. I feel sorry for David but it's the best choice for Series 2. To be honest, too many US based shows end up with the 'everything is now great…except this one…