diff --git a/ui/test/ifql/components/FuncSelector.test.tsx b/ui/test/ifql/components/FuncSelector.test.tsx
index b6c9536e84..8218876099 100644
--- a/ui/test/ifql/components/FuncSelector.test.tsx
+++ b/ui/test/ifql/components/FuncSelector.test.tsx
@@ -133,6 +133,7 @@ describe('IFQL.Components.FuncsButton', () => {
           const onAddNode = jest.fn()
           const {wrapper, props} = setup({onAddNode})
           const [, func2] = props.funcs
+          const {expressionID} = props
 
           const dropdownButton = wrapper.find('button')
           dropdownButton.simulate('click')
@@ -147,7 +148,7 @@ describe('IFQL.Components.FuncsButton', () => {
           input.simulate('keyDown', {key: 'ArrowDown'})
           input.simulate('keyDown', {key: 'Enter'})
 
-          expect(onAddNode).toHaveBeenCalledWith(func2)
+          expect(onAddNode).toHaveBeenCalledWith(func2, expressionID)
         })
       })
     })