diff --git a/jupyter_rtc_core/kernels/websocket_connection.py b/jupyter_rtc_core/kernels/websocket_connection.py
index b34e0ed..4f3b732 100644
--- a/jupyter_rtc_core/kernels/websocket_connection.py
+++ b/jupyter_rtc_core/kernels/websocket_connection.py
@@ -23,7 +23,7 @@ async def connect(self):
         self.kernel_manager.broadcast_state()
         self.log.info("Kernel websocket is now listening to kernel.")
 
-    async def disconnect(self):
+    def disconnect(self):
         self.kernel_manager.main_client.remove_listener(self.handle_outgoing_message)
 
     def handle_incoming_message(self, incoming_msg):
diff --git a/jupyter_rtc_core/session_manager.py b/jupyter_rtc_core/session_manager.py
index b72f023..fce54aa 100644
--- a/jupyter_rtc_core/session_manager.py
+++ b/jupyter_rtc_core/session_manager.py
@@ -86,8 +86,8 @@ async def delete_session(self, session_id):
         Deletes the session and disconnects the yroom from the kernel client.
         """
         session = await self.get_session(session_id=session_id)
-        kernel_id, path = session["kernel_id"], session["path"]
-        yroom = self.get_yroom(path)
+        kernel_id, path, type = session["kernel"]["id"], session["path"], session["type"]
+        yroom = self.get_yroom(path, type)
         kernel_client = self.get_kernel_client(kernel_id)
-        kernel_client.remove_yroom(yroom)
+        await kernel_client.remove_yroom(yroom)
         await super().delete_session(session_id)
\ No newline at end of file