Add a missing import + pickling protocol.