diff --git a/test/index.js b/test/index.js
index 3b51203..ee25781 100644
--- a/test/index.js
+++ b/test/index.js
@@ -57,7 +57,7 @@ describe('babel-plugin-add-module-exports', () => {
   testCases.forEach((testCase) =>
     it(`should ${testCase.name}`, () =>
       testPlugin(testCase.code, {
-        presets: ['env'],
+        presets: [['env', testCase.env]],
         plugins: [
           'transform-export-extensions', // use export-from syntax
           ['./src/index.js', testCase.options]
diff --git a/test/spec.js b/test/spec.js
index 6b23673..f8c8c6c 100644
--- a/test/spec.js
+++ b/test/spec.js
@@ -1,16 +1,42 @@
 module.exports = [
   {
     name: 'export default to module.exports if only export default',
-    options: {},
     code: 'export default "default-entry"',
     expected: {
       module: 'default-entry',
       exports: 'default-entry'
     }
   },
+  // TODO:
+  // {
+  //   name: 'export default to module.exports if only export default using amd',
+  //   code: 'export default "default-entry"',
+  //   env: {modules: 'amd'},
+  //   expected: {
+  //     module: 'default-entry',
+  //     exports: 'default-entry'
+  //   }
+  // },
+  // {
+  //   name: 'export default to module.exports if only export default using umd',
+  //   code: 'export default "default-entry"',
+  //   env: {modules: 'umd'},
+  //   expected: {
+  //     module: 'default-entry',
+  //     exports: 'default-entry'
+  //   }
+  // },
+  // {
+  //   name: 'export default to module.exports if only export default using systemjs',
+  //   code: 'export default "default-entry"',
+  //   env: {modules: 'systemjs'},
+  //   expected: {
+  //     module: 'default-entry',
+  //     exports: 'default-entry'
+  //   }
+  // },
   {
     name: 'export other entries to module.exports if no default entry',
-    options: {},
     code: 'export const other1 = "entry1"; export const other2 = "entry2"',
     expected: {
       module: {
@@ -25,7 +51,6 @@ module.exports = [
   },
   {
     name: 'not export default to module.exports if export multiple entries',
-    options: {},
     code: 'export default "default-entry"; export const other = "other-entry"',
     expected: {
       module: {
@@ -40,7 +65,6 @@ module.exports = [
   },
   {
     name: 'export a function as default entry',
-    options: {},
     code: 'export default () => "default-entry"',
     expected: {
       module: () => 'default-entry',
@@ -49,7 +73,6 @@ module.exports = [
   },
   {
     name: 'export default function entry with other entries',
-    options: {},
     code: 'export default () => "default-entry"; export const other = "other-entry"',
     expected: {
       module: {
@@ -64,7 +87,6 @@ module.exports = [
   },
   {
     name: 'not override default object with other export entries',
-    options: {},
     code: 'export default { value: 1 }; export const value = 2',
     expected: {
       module: {
@@ -81,7 +103,6 @@ module.exports = [
     // even be compatible with wrong behavior: https://github.com/babel/babel/issues/2212#issuecomment-131110500
     // name: 'allow the wrong default export as like in Babel 5',
     name: 'follow the Babel@5 behavior (end of #4)',
-    options: {},
     code: 'export default { name: "test", version: "0.0.1" }',
     expected: {
       module: {
@@ -96,7 +117,6 @@ module.exports = [
   },
   {
     name: 'export default using transform-export-extensions (#11)',
-    options: {},
     code: "export default from './fixtures/issue011.js'",
     expected: {
       module: 'this is file',
@@ -105,7 +125,6 @@ module.exports = [
   },
   {
     name: 'export default and named declarations (#30)',
-    options: {},
     code: 'const foo = 1;const BAR = 2;export { foo as default, BAR }',
     expected: {
       module: {