How about after the boys are reunited in heaven, heaven starts glitching? They investigate and learn that the individual heavens are failing. They immediately rush to check on loved ones and find that not only has Sam's son ed away but he didn't make it to his appointed heaven. The boys get to work and do what they do best.