diff --git a/packages/flutter_tools/test/general.shard/base/io_test.dart b/packages/flutter_tools/test/general.shard/base/io_test.dart
index a1eabf2b11..c6778d0a24 100644
--- a/packages/flutter_tools/test/general.shard/base/io_test.dart
+++ b/packages/flutter_tools/test/general.shard/base/io_test.dart
@@ -74,10 +74,13 @@ void main() {
   });
 
   test('exit does not throw a StateError if overriden', () {
-    setExitFunctionForTests((int value) {});
+    try {
+      setExitFunctionForTests((int value) {});
 
-    expect(() => exit(0), returnsNormally);
-    restoreExitFunction();
+      expect(() => exit(0), returnsNormally);
+    } finally {
+      restoreExitFunction();
+    }
   });
 
   test('test_api defines the Declarer in a known place', () {