I don't think we need the `__reduce__` here anymore?