diff --git a/dvc/commands/dataset.py b/dvc/commands/dataset.py
index ee299f2430..ece8497286 100644
--- a/dvc/commands/dataset.py
+++ b/dvc/commands/dataset.py
@@ -4,7 +4,6 @@
 from dvc.cli.command import CmdBase
 from dvc.cli.utils import append_doc_link
 from dvc.log import logger
-from dvc.repo.datasets import URLDatasetLock
 
 if TYPE_CHECKING:
     from rich.text import Text
@@ -74,7 +73,7 @@ def run(self):
 class CmdDatasetUpdate(CmdBase):
     def display(self, dataset: "Dataset", new: "Dataset"):
         from dvc.commands.checkout import log_changes
-        from dvc.repo.datasets import DVCDatasetLock, DVCXDatasetLock
+        from dvc.repo.datasets import DVCDatasetLock, DVCXDatasetLock, URLDatasetLock
         from dvc.ui import ui
 
         if not dataset.lock: