Update rewire api usageadd-rewire-plugin

2 files changed, 4 insertions, 4 deletions

diff --git a/spec/javascripts/comment_type_toggle_spec.js b/spec/javascripts/comment_type_toggle_spec.js
index 10187085733..2e7343aeb0d 100644
--- a/spec/javascripts/comment_type_toggle_spec.js
+++ b/spec/javascripts/comment_type_toggle_spec.js
@@ -1,4 +1,4 @@
-import CommentTypeToggle, { __RewireAPI__ as CommentTypeToggleRewire } from '~/comment_type_toggle';
+import CommentTypeToggle from '~/comment_type_toggle';
 import InputSetter from '~/droplab/plugins/input_setter';
 
 describe('CommentTypeToggle', function () {
@@ -60,7 +60,7 @@ describe('CommentTypeToggle', function () {
       this.DropLab = jasmine.createSpy('DropLab').and.returnValue(this.droplab);
       spyOn(this.commentTypeToggle, 'setConfig').and.returnValue(this.config);
 
-      CommentTypeToggleRewire.__set__('DropLab', this.DropLab);
+      CommentTypeToggle.__Rewire__('DropLab', this.DropLab);
 
       CommentTypeToggle.prototype.initDroplab.call(this.commentTypeToggle);
     });
diff --git a/spec/javascripts/droplab/hook_spec.js b/spec/javascripts/droplab/hook_spec.js
index a81bd54ac3d..e4625d7e219 100644
--- a/spec/javascripts/droplab/hook_spec.js
+++ b/spec/javascripts/droplab/hook_spec.js
@@ -1,6 +1,6 @@
 /* eslint-disable */
 
-import Hook, { __RewireAPI__ as HookRewire } from '~/droplab/hook';
+import Hook from '~/droplab/hook';
 
 describe('Hook', function () {
   describe('class constructor', function () {
@@ -12,7 +12,7 @@ describe('Hook', function () {
       this.dropdown = {};
 
       this.DropDown = jasmine.createSpy('DropDown').and.returnValue(this.dropdown);
-      HookRewire.__set__('DropDown', this.DropDown);
+      Hook.__Rewire__('DropDown', this.DropDown);
 
       this.hook = new Hook(this.trigger, this.list, this.plugins, this.config);
     });