diff --git a/main/CMakeLists.txt b/main/CMakeLists.txt
index c625f62..2c68e4d 100644
--- a/main/CMakeLists.txt
+++ b/main/CMakeLists.txt
@@ -1,2 +1,2 @@
-idf_component_register(SRCS "service_discovery.c" "app_nvs.c" "apds_9960.c" "pca9555.c" "i2c.c" "asr_pro.c" "ci_03t.c" "ui_input.c" "ambient_light.c" "temperature.c" "embedded_display.c" "mqtt.c" "main.c" "wifi.c" "light.c" "mqtt.c" "led_strip_encoder/led_strip_encoder.c"
+idf_component_register(SRCS "udp_server.c" "service_discovery.c" "app_nvs.c" "apds_9960.c" "pca9555.c" "i2c.c" "asr_pro.c" "ci_03t.c" "ui_input.c" "ambient_light.c" "temperature.c" "embedded_display.c" "mqtt.c" "main.c" "wifi.c" "light.c" "mqtt.c" "led_strip_encoder/led_strip_encoder.c"
                     INCLUDE_DIRS ".")
\ No newline at end of file
diff --git a/main/udp_server.c b/main/udp_server.c
index e31d6fe..c411679 100644
--- a/main/udp_server.c
+++ b/main/udp_server.c
@@ -10,6 +10,7 @@
 #include "esp_wifi.h"
 #include "freertos/FreeRTOS.h"
 #include "freertos/task.h"
+#include "light.c"
 #include "lwip/err.h"
 #include "lwip/sockets.h"
 #include "lwip/sys.h"
@@ -103,6 +104,8 @@ static void udp_server_task(void *pvParameters) {
             ESP_LOGE(UDP_SERVER_TAG,
                      "Error occurred during sending: errno %d. sock: %d", errno,
                      sock);
+            shutdown(sock, 0);
+            close(sock);
             break;
           }
         }