diff --git a/index.js b/index.js
index 2d2c94f..6946e9d 100644
--- a/index.js
+++ b/index.js
@@ -1,9 +1,7 @@
-module.exports = (variants) => {
-  return function ({ addUtilities }) {
-    addUtilities({
-      '.test': {
-        display: 'block'
-      }
-    }, variants)
-  }
+module.exports = (variants) => ({ addUtilities }) => {
+  addUtilities({
+    '.test': {
+      display: 'block'
+    }
+  }, variants)
 }