diff --git a/tests_integration/runPrettier.js b/tests_integration/runPrettier.js
index 2782dfc24..ea1fe0e40 100644
--- a/tests_integration/runPrettier.js
+++ b/tests_integration/runPrettier.js
@@ -62,15 +62,7 @@ function runPrettier(dir, args, options) {
     process.argv = originalArgv;
     process.exitCode = originalExitCode;
     process.stdin.isTTY = originalStdinIsTTY;
-
-    // TODO: use `jest.restoreAllMocks()` once facebook/jest#4436 fixed
-    spiedProcessExit.mockRestore();
-    spiedStdoutWrite.mockRestore();
-    spiedStderrWrite.mockRestore();
-    spiedConsoleLog.mockRestore();
-    spiedConsoleWarn.mockRestore();
-    spiedConsoleError.mockRestore();
-    spiedFsWriteFileSync.mockRestore();
+    jest.restoreAllMocks();
   }
 
   return { status, stdout, stderr, write };