diff --git a/src/wsio_handler.erl b/src/wsio_handler.erl
index affaa98..e5ee29f 100644
--- a/src/wsio_handler.erl
+++ b/src/wsio_handler.erl
@@ -42,7 +42,7 @@ websocket_info({send, <<Bin/binary>>}, S) ->
 websocket_info({send, List}, S) when is_list(List) ->
   {List, S};
 websocket_info({'DOWN', _, process, PID, _}, #{wsbus := PID} = S) ->
-  {ok, S#{wsbus => nil, ref_wsbus => st:lookup(wsbus)}};
+  {[close], S};
 websocket_info({_, {send, _} = Msg}, S) ->
   websocket_info(Msg, S);
 websocket_info({_, {Ref, {error, notfound}}}, #{ref_wsbus := Ref} = S) ->