Yep that went into the Trump calculus. Ohio and PA, and likely Wisconsin.
You are right. Trumps pick had more to do with electoral college votes than anything.