I know the author of the original short story didn't like this version of the ending, but I prefer it.
Correct!
from wikipedia: "Button, Button" was first published in
Playboy, June 1970. The story was republished as part of a collection of Matheson's short stories.
In the original short story, the plot is resolved differently. Norma presses the button, and receives the money—after her husband dies in a train incident, where he is pushed onto the tracks. The money is the no-fault insurance settlement, which is $50,000 instead of the $200,000 in the
Twilight Zone episode. A despondent Norma asks the stranger why her husband was the one who was killed. The stranger replies, "Do you really think you knew your husband?"
Matheson strongly disapproved of the
Twilight Zone version, especially the new ending, and used his pseudonym Logan Swanson for the teleplay.
Usually when a filmed version of a written work differs significantly from the original text, both versions are not written by the same author.
I agree with you that the
The Twilight Zone ending is better.