Fix broken tests in pyzmq>14