diff --git a/tcp.py b/tcp.py
index 65467af..cbd555d 100644
--- a/tcp.py
+++ b/tcp.py
@@ -15,9 +15,7 @@ def __call__(ctx):
 
         @untwisted.call
         class factory(protocol.ClientFactory):
-
-          # Extend protocol.Protocol for .connectionLost()
-          class protocol(protocol.Protocol):
+          class protocol:
             def __init__(ctx):
               ctx.connectionLost = promise.promise()
 
@@ -56,9 +54,7 @@ def listen(port, interface=''):
 
   @untwisted.call
   class factory(protocol.Factory):
-
-    # Extend protocol.Protocol for .connectionLost()
-    class protocol(protocol.Protocol):
+    class protocol:
       def __init__(ctx):
         ctx.connectionLost = promise.promise()