diff --git a/src/pl_bolts/callbacks/knn_online.py b/src/pl_bolts/callbacks/knn_online.py
index 0d0cb8896..9c392a39e 100644
--- a/src/pl_bolts/callbacks/knn_online.py
+++ b/src/pl_bolts/callbacks/knn_online.py
@@ -2,7 +2,7 @@
 
 import torch
 from lightning import Callback, LightningModule, Trainer
-from lightning.accelerators import Accelerator
+from lightning.pytorch.accelerators import Accelerator
 from pl_bolts.utils.stability import under_review
 from torch import Tensor
 from torch.nn import functional as F  # noqa: N812
diff --git a/src/pl_bolts/utils/_dependency.py b/src/pl_bolts/utils/_dependency.py
index 32571c478..81ce2d912 100644
--- a/src/pl_bolts/utils/_dependency.py
+++ b/src/pl_bolts/utils/_dependency.py
@@ -2,10 +2,7 @@
 import os
 from typing import Any, Callable
 
-from lightning.fabric.utilities.core.imports import (
-    ModuleAvailableCache,
-    RequirementCache,
-)
+from lightning_utilities.core.imports import ModuleAvailableCache, RequirementCache
 
 
 # ToDo: replace with utils wrapper after 0.10 is released