I haven't seen AoI in a while, but I remember being very happy with it up until the Amsterdam chase sequence. I thought it was one of the better constructed stories in the Fifth Doctor's run, until it fell apart in that last episode. That chase sequence did have the shots of Omega realizing he'd finally returned to the real world and didn't need to be evil anymore, and I think the story would have been better if had ended with him choosing to follow that path, rather than having circumstances force him to flee and ultimately be destroyed by the Doctor.