1 files changed, 6 insertions, 6 deletions
diff --git a/openecomp-ui/test/nfvo-components/panel/VersionController/versionController.test.js b/openecomp-ui/test/nfvo-components/panel/VersionController/versionController.test.js
index 7d4d57eb35..e14e9b76f5 100644
--- a/openecomp-ui/test/nfvo-components/panel/VersionController/versionController.test.js
+++ b/openecomp-ui/test/nfvo-components/panel/VersionController/versionController.test.js
@@ -16,7 +16,7 @@
 
 
 import React from 'react';
-
+import ReactDOMServer from 'react-dom/server';
 import TestUtils from 'react-addons-test-utils';
 import {mount} from 'enzyme';
 import VersionController from 'nfvo-components/panel/versionController/VersionController.jsx';
@@ -131,21 +131,21 @@ describe('versionController UI Component', () => {
 		let callVCActionProps = { ...props, version: '1.0', callVCAction: function(){} };
 		let versionController = mount(<VersionController isCheckedOut={true} status={statusEnum.CHECK_OUT_STATUS} {...callVCActionProps} />);
 		let elem = versionController.find('[data-test-id="vc-checkout-btn"]');
-		let svgIcon = versionController.find('.versionControllerLockClosed');
+
 		expect(elem).toBeTruthy();
 		expect(elem.length).toEqual(1);
-		expect(svgIcon.hasClass('disabled')).toBe(true);
+		expect(elem.find('.svg-icon').length).toEqual(1);
+		expect(elem.find('.svg-icon').hasClass('disabled')).toBe(true);
 	});
 
 	it('Doesn\'t show the checkout button', () => {
 		let callVCActionProps = { ...props, version: '1.0', callVCAction: function(){} };
 		let versionController = mount(<VersionController isCheckedOut={false} status={statusEnum.CHECK_IN_STATUS} {...callVCActionProps} />);
 		let elem = versionController.find('[data-test-id="vc-checkout-btn"]');
-		let svgIcon = versionController.find('.versionControllerLockClosed');
-
 		expect(elem).toBeTruthy();
 		expect(elem.length).toBe(1);
-		expect(svgIcon.hasClass('disabled')).toBe(true);
+		expect(elem.find('.svg-icon').length).toEqual(1);
+		expect(elem.find('.svg-icon').hasClass('disabled')).toBe(true);
 
 	});