diff --git a/examples/lte_awsiot/subscribe_publish_sample.c b/examples/lte_awsiot/subscribe_publish_sample.c
index eb268baef..4a8302459 100644
--- a/examples/lte_awsiot/subscribe_publish_sample.c
+++ b/examples/lte_awsiot/subscribe_publish_sample.c
@@ -131,7 +131,7 @@ static void parseInputArgsForConnectParams(int argc, char **argv) {
 
 }
 
-int main(int argc, char FAR **argv)
+static int awsiot_main(int argc, FAR char *argv[])
 {
 	bool infinitePublishFlag = true;
 
@@ -145,9 +145,6 @@ int main(int argc, char FAR **argv)
 
 	IoT_Error_t rc = FAILURE;
 
-	if (app_awsiot_connect_to_lte())
-		return ERROR;
-
 	AWS_IoT_Client client;
 	IoT_Client_Init_Params mqttInitParams = iotClientInitParamsDefault;
 	IoT_Client_Connect_Params connectParams = iotClientConnectParamsDefault;
@@ -276,7 +273,22 @@ int main(int argc, char FAR **argv)
 	}
 
 	aws_iot_mqtt_disconnect(&client);
-	app_awsiot_disconnect_from_lte();
 
 	return rc;
 }
+
+int main(int argc, FAR char *argv[])
+{
+  int ret = 0;
+
+  if (app_awsiot_connect_to_lte())
+    {
+      return ERROR;
+    }
+
+  ret = awsiot_main(argc, argv);
+
+  app_awsiot_disconnect_from_lte();
+
+  return ret;
+}