diff --git a/src/ejabberd_socket.erl b/src/ejabberd_socket.erl
index e26fc8652..e83fd3f31 100644
--- a/src/ejabberd_socket.erl
+++ b/src/ejabberd_socket.erl
@@ -227,8 +227,8 @@ get_transport(#socket_state{sockmod = SockMod,
 	fast_tls -> tls;
 	ezlib ->
 	    case ezlib:get_sockmod(Socket) of
-		tcp -> tcp_zlib;
-		tls -> tls_zlib
+		gen_tcp -> tcp_zlib;
+		fast_tls -> tls_zlib
 	    end;
 	ejabberd_bosh -> http_bind;
 	ejabberd_http_bind -> http_bind;