diff --git a/jest.json b/jest.json
index 1f61a01359..82f2a65dfc 100644
--- a/jest.json
+++ b/jest.json
@@ -7,7 +7,9 @@
   "bail": true,
   "resetMocks": true,
   "resetModules": true,
-  "setupTestFrameworkScriptFile": "<rootDir>/tests/setup.js",
+  "setupFilesAfterEnv": [
+    "<rootDir>/tests/setup.js"
+  ],
   "moduleFileExtensions": [
     "js",
     "jsx",
diff --git a/package.json b/package.json
index 9a4750ff3b..04f6eed655 100644
--- a/package.json
+++ b/package.json
@@ -61,6 +61,7 @@
     "update-electron-app": "^1.3.0"
   },
   "devDependencies": {
+    "@babel/core": "^7.4.0",
     "@electron-forge/cli": "^6.0.0-beta.34",
     "@electron-forge/maker-deb": "^6.0.0-beta.34",
     "@electron-forge/maker-rpm": "^6.0.0-beta.34",
diff --git a/wallaby.js b/wallaby.js
index efd8288129..3fa196bd16 100644
--- a/wallaby.js
+++ b/wallaby.js
@@ -21,12 +21,10 @@ module.exports = (wallaby) => ({
 
   testFramework: 'jest',
 
-  // Enable mock hoisting as same as ts-jest does
-  // (https://github.com/kulshekhar/ts-jest#supports-automatic-of-jestmock-calls)
   preprocessors: {
-    '**/*.js?(x)': (file) => require('babel-core').transform(
+    '**/*.js?(x)': file => require('@babel/core').transform(
       file.content,
-      { sourceMaps: true, filename: file.path, presets: ['babel-preset-jest'] })
+      {sourceMap: true, filename: file.path, presets: [require('babel-preset-jest')]})
   },
 
   workers: {