Airport for Grand Tetons & Yellowstone

My friends and I are planning a trip to the Grand Tetons and Yellowstone in June. We were going to fly into Jackson Hole spend two days in the Grand Tetons, then drive to Yellowstone and spend two days there. So we were thinking of flying out of the Yellowstone airport.

Would it be just as easy to fly back out of Jackson Hole? It's hard to tell on Google Maps right now because it shows road closures with it being winter so we can't get an accurate idea on how far away the drives from the airports to each park is.