diff --git a/dvc/data/checkout.py b/dvc/data/checkout.py
index 756f2832d..e935acb01 100644
--- a/dvc/data/checkout.py
+++ b/dvc/data/checkout.py
@@ -1,12 +1,12 @@
 import logging
 from itertools import chain
+from typing import Optional
 
 from dvc import prompt
 from dvc.exceptions import CacheLinkError, CheckoutError, ConfirmRemoveError
 from dvc.ignore import DvcIgnoreFilter
 from dvc.objects.fs._callback import FsspecCallback
 from dvc.objects.fs.generic import test_links, transfer
-from dvc.types import Optional
 
 from .diff import ROOT
 from .diff import diff as odiff
diff --git a/dvc/data/db/index.py b/dvc/data/db/index.py
index 7c8c5623f..2063ed46a 100644
--- a/dvc/data/db/index.py
+++ b/dvc/data/db/index.py
@@ -6,7 +6,7 @@ from typing import TYPE_CHECKING, Iterable, Set
 from dvc.objects.errors import ObjectDBError
 
 if TYPE_CHECKING:
-    from dvc.types import StrPath
+    from dvc.objects.fs.base import AnyFSPath
 
 logger = logging.getLogger(__name__)
 
@@ -15,7 +15,7 @@ class ObjectDBIndexBase(ABC):
     @abstractmethod
     def __init__(
         self,
-        tmp_dir: "StrPath",
+        tmp_dir: "AnyFSPath",
         name: str,
     ):
         pass
@@ -53,7 +53,7 @@ class ObjectDBIndexNoop(ObjectDBIndexBase):
 
     def __init__(
         self,
-        tmp_dir: "StrPath",
+        tmp_dir: "AnyFSPath",
         name: str,
     ):  # pylint: disable=super-init-not-called
         pass
@@ -85,7 +85,7 @@ class ObjectDBIndex(ObjectDBIndexBase):
 
     def __init__(
         self,
-        tmp_dir: "StrPath",
+        tmp_dir: "AnyFSPath",
         name: str,
     ):  # pylint: disable=super-init-not-called
         from dvc.objects.cache import Cache, Index