diff --git a/src/emqttd_client.erl b/src/emqttd_client.erl
index 88f14c2d6..be08b6fed 100644
--- a/src/emqttd_client.erl
+++ b/src/emqttd_client.erl
@@ -112,7 +112,7 @@ do_init(Conn, Env, Peername) ->
     RateLimit = get_value(rate_limit, Conn:opts()),
     PacketSize = get_value(max_packet_size, Env, ?MAX_PACKET_SIZE),
     Parser = emqttd_parser:initial_state(PacketSize),
-    ProtoState = emqttd_protocol:init(Peername, SendFun, Env),
+    ProtoState = emqttd_protocol:init(Conn, Peername, SendFun, Env),
     EnableStats = get_value(client_enable_stats, Env, false),
     ForceGcCount = emqttd_gc:conn_max_gc_count(),
     State = run_socket(#client_state{connection     = Conn,
diff --git a/src/emqttd_ws_client.erl b/src/emqttd_ws_client.erl
index 4beb7bc40..68b00e501 100644
--- a/src/emqttd_ws_client.erl
+++ b/src/emqttd_ws_client.erl
@@ -93,7 +93,7 @@ init([Env, WsPid, Req, ReplyChannel]) ->
     Headers = mochiweb_headers:to_list(
                 mochiweb_request:get(headers, Req)),
     Conn = Req:get(connection),
-    ProtoState = emqttd_protocol:init(Peername, send_fun(ReplyChannel),
+    ProtoState = emqttd_protocol:init(Conn, Peername, send_fun(ReplyChannel),
                                       [{ws_initial_headers, Headers} | Env]),
     IdleTimeout = get_value(client_idle_timeout, Env, 30000),
     EnableStats = get_value(client_enable_stats, Env, false),