I like this version, but I agree. The one they ended with is much more fitting. Mike's sitting there all alone at the table in the flashback and then he's all alone at the compound. A much colder ending.