diff --git a/app/soapbox/components/__tests__/timeline_queue_button_header.test.js b/app/soapbox/components/__tests__/timeline_queue_button_header.test.js
index 6874452ae..155179551 100644
--- a/app/soapbox/components/__tests__/timeline_queue_button_header.test.js
+++ b/app/soapbox/components/__tests__/timeline_queue_button_header.test.js
@@ -1,3 +1,4 @@
+import { fromJS } from 'immutable';
 import React from 'react';
 import { defineMessages } from 'react-intl';
 
@@ -14,9 +15,11 @@ describe('<TimelineQueueButtonHeader />', () => {
       <TimelineQueueButtonHeader
         key='timeline-queue-button-header'
         onClick={() => {}} // eslint-disable-line react/jsx-no-bind
-        count={0}
+        timelineId='home'
         message={messages.queue}
       />,
+      undefined,
+      { timelines: fromJS({ home: { totalQueuedItemsCount: 0 } }) },
     );
     expect(screen.queryAllByRole('link')).toHaveLength(0);
 
@@ -24,9 +27,11 @@ describe('<TimelineQueueButtonHeader />', () => {
       <TimelineQueueButtonHeader
         key='timeline-queue-button-header'
         onClick={() => {}} // eslint-disable-line react/jsx-no-bind
-        count={1}
+        timelineId='home'
         message={messages.queue}
       />,
+      undefined,
+      { timelines: fromJS({ home: { totalQueuedItemsCount: 1 } }) },
     );
     expect(screen.getByText('Click to see 1 new post', { hidden: true })).toBeInTheDocument();
 
@@ -34,9 +39,11 @@ describe('<TimelineQueueButtonHeader />', () => {
       <TimelineQueueButtonHeader
         key='timeline-queue-button-header'
         onClick={() => {}} // eslint-disable-line react/jsx-no-bind
-        count={9999999}
+        timelineId='home'
         message={messages.queue}
       />,
+      undefined,
+      { timelines: fromJS({ home: { totalQueuedItemsCount: 9999999 } }) },
     );
     expect(screen.getByText('Click to see 9999999 new posts', { hidden: true })).toBeInTheDocument();
   });