diff --git a/include/hass.hpp b/include/hass.hpp
index 16f74c6..7ba7415 100644
--- a/include/hass.hpp
+++ b/include/hass.hpp
@@ -13,7 +13,7 @@
 #define HASS_ENTITY "sensor.gniazdko_komputer_moc"
 
 int hass_connect();
-void hass_poll();
+void hass_service(int timeout_ms = 0);
 float hass_get();
 void hass_disconnect();
 
diff --git a/src/hass.cpp b/src/hass.cpp
index ce854f3..d9d697f 100644
--- a/src/hass.cpp
+++ b/src/hass.cpp
@@ -139,9 +139,9 @@ int hass_connect() {
     return 0;
 }
 
-void hass_poll() {
-    lws_service(context, 0);
+void hass_service(int timeout_ms) {
     lws_callback_on_writable(websock);
+    lws_service(context, timeout_ms);
 }
 
 float hass_get() {