Realizing Diane can't remember what happened to Kurt, things start to get tense. Viewers might recall that Season 3 ended with a scary cliffhanger: After Diane goes against her "book club" (but actually, an underground faction of women trying to swing the election in favor of the Democrats), they decide to get revenge on her by swatting her (i.e., deceiving an emergency service into sending an emergency police response to another person's address).

We all saw the S.W.A.T. team preparing to enter Diane and Kurt's apartment in the finale, but the episode ends before we can see what happens — so it wasn't far-fetched to assume that Kurt, a ballistics expert and proud gun owner, may not have survived the incident.

Now, back to Season 4. Once Diane realizes she hasn't seen Kurt in the premiere, she sets out to look for him and eventually finds him seated at a table in the middle of a forest with his face shrouded in darkness.

The little details suggest that Kurt died during the swatting and that Diane was having some sort of grief-induced dream. She tells Kurt that their relationship is all that matters to her, regardless of who's president. Kurt responds by telling Diane to come back.

This jolts Diane's memory of the swatting, and again, she believes Kurt may have died. But it's a huge relief when Diane opens her eyes and finds Kurt kneeling by her side in the bedroom.

It turns out that Diane had fallen and hit her head when the S.W.A.T. team entered her home, and passed out. Meaning that the entire alternate reality of Hillary as president was just a dream. 

And Kurt is fine, thank goodness. We just hope we get to see a little more of him, and catch up with Diane's life outside the office this season.

Catch all new episodes of The Good Fight, Thursdays on CBS All Access.