Add missing include.
This code needs to become a wrapper class for simplicity.

Otherwise there's too much typing all the time.