diff --git a/src/Network/GemServ.hs b/src/Network/GemServ.hs
index 64b7263..c6a6f7c 100644
--- a/src/Network/GemServ.hs
+++ b/src/Network/GemServ.hs
@@ -59,7 +59,7 @@ import Data.X509 (Certificate, CertificateChain (..), getSigned, signedObject)
 import qualified Network.Socket as S
 import Network.TLS (ServerParams, onClientCertificate, serverHooks)
 import System.IO.Streams.TCP (bindAndListen)
-import System.IO.Streams.TLS (TLSConnection, accept)
+import System.IO.Streams.TLS (accept)
 
 import Network.GemServ.Internal
 import Network.GemServ.Types
@@ -209,7 +209,7 @@ adjustServerParams certRef params = let
   in params { serverHooks = hooks' }
 
 runConnection
-  :: TLSConnection
+  :: C.Connection a
   -> GemHandler
   -> Maybe Certificate
   -> IO ()