diff --git a/src/ejabberd_c2s.erl b/src/ejabberd_c2s.erl
index 33d76b07c..2cdf205cc 100644
--- a/src/ejabberd_c2s.erl
+++ b/src/ejabberd_c2s.erl
@@ -550,8 +550,7 @@ wait_for_auth({xmlstreamelement, El}, StateData) ->
 				[StateData#state.socket,
 				 jlib:jid_to_string(JID), AuthModule]),
 			SID = {now(), self()},
-			Conn = (StateData#state.sockmod):get_conn_type(
-				    StateData#state.socket),
+			Conn = get_conn_type(StateData),
 			Info = [{ip, StateData#state.ip}, {conn, Conn},
 				    {auth_module, AuthModule}],
                         Res = jlib:make_result_iq_reply(