diff --git a/stories/Console.stories.tsx b/stories/Console.stories.tsx
index 423af4b..2b76f95 100644
--- a/stories/Console.stories.tsx
+++ b/stories/Console.stories.tsx
@@ -1186,30 +1186,11 @@ export const Working = () => {
       >
         <SessionsList>
           <NewSessionButton />
-          <SessionGroups>
-            {groups =>
-              groups.map(({ heading, sessions }) => (
-                <SessionsGroup heading={heading} key={heading}>
-                  {sessions.map(s => (
-                    <SessionListItem key={s.id} session={s} />
-                  ))}
-                </SessionsGroup>
-              ))
-            }
-          </SessionGroups>
+          <SessionGroups />
         </SessionsList>
         <SessionMessagePanel>
           <SessionMessagesHeader />
-          <SessionMessages>
-            {conversations =>
-              conversations.map(conversation => (
-                <SessionMessage
-                  key={conversation.id}
-                  conversation={conversation}
-                />
-              ))
-            }
-          </SessionMessages>
+          <SessionMessages />
           <ChatInput />
         </SessionMessagePanel>
       </Chat>