2 files changed, 5 insertions, 6 deletions
diff --git a/src/client.c b/src/client.c
index 34062ed..3aba76c 100644
--- a/src/client.c
+++ b/src/client.c
@@ -116,7 +116,7 @@ static MQTT_INLINE int MqttClientInflightMessageCount(MqttClient *client)
     return inMessagesCount + outMessagesCount - queued;
 }
 
-MqttClient *MqttClientNew(const char *clientId, int cleanSession)
+MqttClient *MqttClientNew(const char *clientId)
 {
     MqttClient *client;
 
@@ -129,8 +129,6 @@ MqttClient *MqttClientNew(const char *clientId, int cleanSession)
 
     client->clientId = bfromcstr(clientId);
 
-    client->cleanSession = cleanSession;
-
     client->stream.sock = -1;
 
     client->retryTimeout = 20;
@@ -216,7 +214,7 @@ void MqttClientSetOnPublish(MqttClient *client,
 }
 
 int MqttClientConnect(MqttClient *client, const char *host, short port,
-                      int keepAlive)
+                      int keepAlive, int cleanSession)
 {
     int sock;
     MqttPacketConnect *packet;
@@ -227,6 +225,7 @@ int MqttClientConnect(MqttClient *client, const char *host, short port,
     client->host = bfromcstr(host);
     client->port = port;
     client->keepAlive = keepAlive;
+    client->cleanSession = cleanSession;
 
     if (keepAlive < 0)
     {
diff --git a/src/mqtt.h b/src/mqtt.h
index 74b1ac0..d7b62c9 100644
--- a/src/mqtt.h
+++ b/src/mqtt.h
@@ -46,7 +46,7 @@ typedef void (*MqttClientOnMessageCallback)(MqttClient *client,
 
 typedef void (*MqttClientOnPublishCallback)(MqttClient *client, int id);
 
-MqttClient *MqttClientNew(const char *clientId, int cleanSession);
+MqttClient *MqttClientNew(const char *clientId);
 
 void MqttClientFree(MqttClient *client);
 
@@ -69,7 +69,7 @@ void MqttClientSetOnPublish(MqttClient *client,
                             MqttClientOnPublishCallback cb);
 
 int MqttClientConnect(MqttClient *client, const char *host, short port,
-                      int keepAlive);
+                      int keepAlive, int cleanSession);
 
 int MqttClientDisconnect(MqttClient *client);