diff --git a/package.json b/package.json
index 8183e339..91592816 100644
--- a/package.json
+++ b/package.json
@@ -12,7 +12,7 @@
   "devDependencies": {
     "alt-search-docs": "1.0.6",
     "babel-cli": "6.6.5",
-    "babel-core": "6.7.4",
+    "babel-core": "6.21.0",
     "babel-eslint": "5.0.0",
     "babel-loader": "6.2.4",
     "babel-plugin-add-module-exports": "^0.1.2",
diff --git a/test/batching-test.js b/test/batching-test.js
index 6ef248ff..37485675 100644
--- a/test/batching-test.js
+++ b/test/batching-test.js
@@ -6,12 +6,13 @@ import sinon from 'sinon'
 import TestUtils from 'react-addons-test-utils'
 import ReactDom from 'react-dom'
 
-// TOOD action was called but not dispatched?
 const Actions = {
   buttonClick() {
     setTimeout(() => {
       this.switchComponent()
     }, 10)
+
+    return null
   },
 
   switchComponent() {