Add missing include.
More graceful flipping.