Add missing import

We use it in the ZMQChannel, I'm not sure how this works without the import