I just saw this episode again the other day, the ending is perfectly fine. That Harlan Ellison was a piece of work.