diff --git a/src/kemal/dsl.cr b/src/kemal/dsl.cr
index 3c8668e..73d9262 100644
--- a/src/kemal/dsl.cr
+++ b/src/kemal/dsl.cr
@@ -6,6 +6,6 @@ HTTP_METHODS = %w(get post put patch delete)
   end
 {% end %}
 
-def ws(path, &block : HTTP::WebSocketHandler::WebSocket -> _)
+def ws(path, &block : HTTP::WebSocket -> _)
   Kemal::WebSocketHandler.new path, &block
 end