Some minor import fixes

Author: knutwannheden

rewrite/rewrite/java/extensions.py

@@ -1,3 +1,4 @@
+from dataclasses import replace
 from typing import Optional, TypeVar, TYPE_CHECKING
 
 from .support_types import J, JRightPadded, JLeftPadded, JContainer, Space

rewrite/rewrite/visitor.py

@@ -3,10 +3,9 @@
 from dataclasses import dataclass
 from typing import Protocol, TypeVar, Optional, Dict, List, Any, cast, Type, ClassVar, TYPE_CHECKING
 
-if TYPE_CHECKING:
-    from .markers import Marker, Markers
-    from .execution import RecipeRunException
-    from .tree import SourceFile, Tree
+from .execution import RecipeRunException
+from .markers import Marker, Markers
+from .tree import SourceFile, Tree
 
 O = TypeVar('O')
 T = TypeVar('T', bound='Tree')