2 files changed, 23 insertions, 13 deletions
diff --git a/slixmpp/features/feature_starttls/stanza.py b/slixmpp/features/feature_starttls/stanza.py
index 5552cf61..70979402 100644
--- a/slixmpp/features/feature_starttls/stanza.py
+++ b/slixmpp/features/feature_starttls/stanza.py
@@ -4,39 +4,47 @@
 # This file is part of Slixmpp.
 # See the file LICENSE for copying permission.
 from slixmpp.xmlstream import StanzaBase, ElementBase
+from typing import Set, ClassVar
 
 
-class STARTTLS(ElementBase):
-
-    """
+class STARTTLS(StanzaBase):
     """
 
+    .. code-block:: xml
+
+         <starttls xmlns='urn:ietf:params:xml:ns:xmpp-tls'/>
+
+    """
     name = 'starttls'
     namespace = 'urn:ietf:params:xml:ns:xmpp-tls'
     interfaces = {'required'}
     plugin_attrib = name
 
     def get_required(self):
-        """
-        """
         return True
 
 
 class Proceed(StanzaBase):
-
-    """
     """
 
+    .. code-block:: xml
+
+        <proceed xmlns='urn:ietf:params:xml:ns:xmpp-tls'/>
+
+    """
     name = 'proceed'
     namespace = 'urn:ietf:params:xml:ns:xmpp-tls'
-    interfaces = set()
+    interfaces: ClassVar[Set[str]] = set()
 
 
 class Failure(StanzaBase):
-
-    """
     """
 
+    .. code-block:: xml
+
+        <failure xmlns='urn:ietf:params:xml:ns:xmpp-tls'/>
+
+    """
     name = 'failure'
     namespace = 'urn:ietf:params:xml:ns:xmpp-tls'
-    interfaces = set()
+    interfaces: ClassVar[Set[str]] = set()
diff --git a/slixmpp/features/feature_starttls/starttls.py b/slixmpp/features/feature_starttls/starttls.py
index fe793a2d..318d4a5e 100644
--- a/slixmpp/features/feature_starttls/starttls.py
+++ b/slixmpp/features/feature_starttls/starttls.py
@@ -12,6 +12,8 @@ from slixmpp.xmlstream.matcher import MatchXPath
 from slixmpp.xmlstream.handler import CoroutineCallback
 from slixmpp.features.feature_starttls import stanza
 
+from typing import ClassVar, Set
+
 
 log = logging.getLogger(__name__)
 
@@ -20,7 +22,7 @@ class FeatureSTARTTLS(BasePlugin):
 
     name = 'feature_starttls'
     description = 'RFC 6120: Stream Feature: STARTTLS'
-    dependencies = set()
+    dependencies: ClassVar[Set[str]] = set()
     stanza = stanza
 
     def plugin_init(self):
@@ -52,7 +54,7 @@ class FeatureSTARTTLS(BasePlugin):
         elif self.xmpp.disable_starttls:
             return False
         else:
-            self.xmpp.send(features['starttls'])
+            self.xmpp.send(stanza.STARTTLS())
             return True
 
     async def _handle_starttls_proceed(self, proceed):