diff --git a/src/util/define-classes.js b/src/util/define-classes.js
index 2bf9d3845..1e14718eb 100644
--- a/src/util/define-classes.js
+++ b/src/util/define-classes.js
@@ -3,9 +3,7 @@ const _ = require('lodash')
 const defineClass = require('./define-class')
 
 module.exports = function defineClasses(classes) {
-    let rules = _.map(classes, function(properties, className) {
+    return _.map(classes, function(properties, className) {
         return defineClass(className, properties)
     })
-
-    return rules
 }
diff --git a/tests/defineClasses.test.js b/tests/defineClasses.test.js
index a986400c9..c79adb011 100644
--- a/tests/defineClasses.test.js
+++ b/tests/defineClasses.test.js
@@ -25,6 +25,7 @@ var inlineFlexHelper = `.inline-flex {
  */
 it('it generates a set of helper classes from a config', () => {
     output = defineClasses(config)
+    expect(output).toBeInstanceOf(Array)
     expect(output[0].toString()).toEqual(flexHelper)
     expect(output[1].toString()).toEqual(inlineFlexHelper)
 })