diff --git a/superset-frontend/src/visualizations/FilterBox/FilterBox.test.jsx b/superset-frontend/src/visualizations/FilterBox/FilterBox.test.jsx
index 4afa29d2fecc..e37a4bf1c69d 100644
--- a/superset-frontend/src/visualizations/FilterBox/FilterBox.test.jsx
+++ b/superset-frontend/src/visualizations/FilterBox/FilterBox.test.jsx
@@ -17,15 +17,13 @@
  * under the License.
  */
 import React from 'react';
-import { shallow } from 'enzyme';
 import { styledMount as mount } from 'spec/helpers/theming';
 import FilterBox from 'src/visualizations/FilterBox/FilterBox';
 import SelectControl from 'src/explore/components/controls/SelectControl';
-import { supersetTheme } from '@superset-ui/core';
 
 describe('FilterBox', () => {
   it('should only add defined non-predefined options to filtersChoices', () => {
-    const wrapper = shallow(
+    const wrapper = mount(
       <FilterBox
         chartId={1001}
         datasource={{ id: 1 }}
@@ -47,11 +45,9 @@ describe('FilterBox', () => {
           },
         ]}
         origSelectedValues={{}}
-        theme={supersetTheme}
       />,
     );
-    const component = wrapper.shallow();
-    const inst = component.instance();
+    const inst = wrapper.find('FilterBox').instance();
     // choose a predefined value
     inst.setState({ selectedValues: { name: ['John'] } });
     expect(inst.props.filtersChoices.name.length).toEqual(2);