Fix import pyzmq