diff --git a/components/esp_websocket_client/CMakeLists.txt b/components/esp_websocket_client/CMakeLists.txt
index 723199ce0..f366a278d 100644
--- a/components/esp_websocket_client/CMakeLists.txt
+++ b/components/esp_websocket_client/CMakeLists.txt
@@ -1,4 +1,3 @@
-set(COMPONENT_SRCS "esp_websocket_client.c")
-set(COMPONENT_ADD_INCLUDEDIRS "include")
-set(COMPONENT_REQUIRES lwip esp-tls tcp_transport nghttp)
-register_component()
+idf_component_register(SRCS "esp_websocket_client.c"
+                    INCLUDE_DIRS "include"
+                    REQUIRES lwip esp-tls tcp_transport nghttp)
diff --git a/examples/protocols/websocket/main/CMakeLists.txt b/examples/protocols/websocket/main/CMakeLists.txt
index caf642155..bff26f108 100644
--- a/examples/protocols/websocket/main/CMakeLists.txt
+++ b/examples/protocols/websocket/main/CMakeLists.txt
@@ -1,4 +1,2 @@
-set(COMPONENT_SRCS "websocket_example.c")
-set(COMPONENT_ADD_INCLUDEDIRS ".")
-
-register_component()
+idf_component_register(SRCS "websocket_example.c"
+                    INCLUDE_DIRS ".")