Yep, see above. Either we make that not break with a `KeyError` by using `get` or simply remove the optional nature of the decorator. Or use classes.