diff --git a/python/testSrc/com/jetbrains/python/PyOptimizeImportsTest.java b/python/testSrc/com/jetbrains/python/PyOptimizeImportsTest.java
index 9346c602c600..668b53d84186 100644
--- a/python/testSrc/com/jetbrains/python/PyOptimizeImportsTest.java
+++ b/python/testSrc/com/jetbrains/python/PyOptimizeImportsTest.java
@@ -23,6 +23,14 @@ import com.jetbrains.python.fixtures.PyTestCase;
  * @author yole
  */
 public class PyOptimizeImportsTest extends PyTestCase {
+
+  @Override
+  protected void setUp() throws Exception {
+    super.setUp();
+    // importsFromTypingUnusedInTypeComments depends on registered TokenSetContributors
+    PythonDialectsTokenSetProvider.reset();
+  }
+
   public void testSimple() {
     doTest();
   }