Are we comfortable raising `KeyError` here or do we want a more helpful error message?