It's ugly but you can fix this with `dataclass_transform`