diff --git a/test/servers/ServersDropdown.test.js b/test/servers/ServersDropdown.test.js
index 3f8013f0..dde12a82 100644
--- a/test/servers/ServersDropdown.test.js
+++ b/test/servers/ServersDropdown.test.js
@@ -2,6 +2,7 @@ import { identity } from 'ramda';
import React from 'react';
import { ServersDropdown } from '../../src/servers/ServersDropdown';
import { shallow } from 'enzyme';
+import { DropdownItem, DropdownToggle } from 'reactstrap';
describe('', () => {
let wrapped;
@@ -13,10 +14,10 @@ describe('', () => {
afterEach(() => wrapped.unmount());
it('contains the list of servers', () => {
- expect(wrapped.find('DropdownItem').length).toEqual(servers.length);
+ expect(wrapped.find(DropdownItem).length).toEqual(servers.length + 2);
});
it('contains a toggle with proper title', () => {
- expect(wrapped.find('DropdownToggle').length).toEqual(1);
+ expect(wrapped.find(DropdownToggle).length).toEqual(1);
});
});