diff --git a/test_autoflake.py b/test_autoflake.py
index 9acdbb3..ad3315a 100755
--- a/test_autoflake.py
+++ b/test_autoflake.py
@@ -928,6 +928,7 @@ def test_fix_code_with_from_and_depth_module(self) -> None:
 from distutils.version import LooseVersion, StrictVersion
 StrictVersion('1.0.0')
 """,
+                remove_all_unused_imports=True,
             ),
         )
 
@@ -941,6 +942,7 @@ def test_fix_code_with_from_and_depth_module(self) -> None:
 from distutils.version import LooseVersion, StrictVersion as version
 version('1.0.0')
 """,
+                remove_all_unused_imports=True,
             ),
         )