diff --git a/src/ui/public/vislib/__tests__/components/color.js b/src/ui/public/vislib/__tests__/components/color.js
index a882d4346b23..7e38078a6609 100644
--- a/src/ui/public/vislib/__tests__/components/color.js
+++ b/src/ui/public/vislib/__tests__/components/color.js
@@ -24,7 +24,7 @@ describe('Vislib Color Module Test Suite', function () {
     var color;
 
     beforeEach(ngMock.module('kibana'));
-    beforeEach(ngMock.inject(function (Private, config) {
+    beforeEach(ngMock.inject((Private, config) => {
       previousConfig = config.get('visualization:colorMapping');
       config.set('visualization:colorMapping', {});
       seedColors = Private(require('ui/vislib/components/color/seed_colors'));
@@ -33,7 +33,7 @@ describe('Vislib Color Module Test Suite', function () {
       color = getColors(arr);
     }));
 
-    afterEach(ngMock.inject(function (config) {
+    afterEach(ngMock.inject((config) => {
       config.set('visualization:colorMapping', previousConfig);
     }));
 
diff --git a/src/ui/public/vislib/components/color/color.js b/src/ui/public/vislib/components/color/color.js
index 522eaf702084..29fe7fac327a 100644
--- a/src/ui/public/vislib/components/color/color.js
+++ b/src/ui/public/vislib/components/color/color.js
@@ -1,5 +1,5 @@
 define(function (require) {
-  return function ColorUtilService(Private, config) {
+  return function ColorUtilService(Private) {
     var _ = require('lodash');
     var mappedColors = Private(require('ui/vislib/components/color/mapped_colors'));