diff --git a/src/mqtt/protocol/mqtt/mqtt_client.c b/src/mqtt/protocol/mqtt/mqtt_client.c
index eb86c3c33..cfa9b5070 100644
--- a/src/mqtt/protocol/mqtt/mqtt_client.c
+++ b/src/mqtt/protocol/mqtt/mqtt_client.c
@@ -330,8 +330,8 @@ mqtt_pipe_recv_msgq_putq(mqtt_pipe_t *p, nni_msg *msg)
 		// 	return;
 		// }
 		// nni_lmq_put(&p->recv_messages, msg);
-		packet_type_t packet_type = nni_mqtt_msg_get_packet_type(msg);
-		if (packet_type == NNG_MQTT_PUBLISH || packet_type == NNG_MQTT_CONNACK)
+		uint8_t type = nni_msg_get_type(msg);
+		if (type == CMD_PUBLISH || type == CMD_CONNACK)
 			conn_param_free(nni_msg_get_conn_param(msg));
 		nni_msg_free(msg);
 	}
diff --git a/src/mqtt/protocol/mqtt/mqttv5_client.c b/src/mqtt/protocol/mqtt/mqttv5_client.c
index aa0c78ba3..ff6747f8a 100644
--- a/src/mqtt/protocol/mqtt/mqttv5_client.c
+++ b/src/mqtt/protocol/mqtt/mqttv5_client.c
@@ -317,9 +317,8 @@ static inline void
 mqtt_pipe_recv_msgq_putq(mqtt_pipe_t *p, nni_msg *msg)
 {
 	if (0 != nni_lmq_put(&p->recv_messages, msg)) {
-		packet_type_t packet_type = nni_mqtt_msg_get_packet_type(msg);
-		if (packet_type == NNG_MQTT_PUBLISH ||
-		    packet_type == NNG_MQTT_CONNACK)
+		uint8_t type = nni_msg_get_type(msg);
+		if (type == CMD_PUBLISH || type == CMD_CONNACK)
 			conn_param_free(nni_msg_get_conn_param(msg));
 		nni_msg_free(msg);
 	}