diff --git a/src/saml2/saml.py b/src/saml2/saml.py index 483611f1c..35b7bd1a4 100644 --- a/src/saml2/saml.py +++ b/src/saml2/saml.py @@ -169,6 +169,13 @@ def set_type(self, typ): except AttributeError: self._extatt[XSI_TYPE] = typ + if typ.startswith('xs:'): + try: + self.extension_attributes['xmlns:xs'] = XS_NAMESPACE + except AttributeError: + self._extatt['xmlns:xs'] = XS_NAMESPACE + + def get_type(self): try: return self.extension_attributes[XSI_TYPE]