diff --git a/requirements.txt b/requirements.txt
index 339b2bbead..c2dbe25984 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,4 +12,4 @@ jsonargparse[signatures] >=3.17.0, <=4.9.0
 click >=7.1.2, <=8.1.3
 protobuf <=3.20.1
 fsspec[http] >=2022.5.0,<=2022.7.1
-lightning-utilities >=0.4.1
+lightning-utilities >=0.3.0
diff --git a/src/flash/core/data/base_viz.py b/src/flash/core/data/base_viz.py
index 3d4f7d5922..9f8c90cae7 100644
--- a/src/flash/core/data/base_viz.py
+++ b/src/flash/core/data/base_viz.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 from typing import Any, Dict, List, Set, Tuple
 
-from lightning_utilities.core.overrides import is_overridden
+from pytorch_lightning.utilities.model_helpers import is_overridden
 
 from flash.core.data.callback import BaseDataFetcher
 from flash.core.data.utils import _CALLBACK_FUNCS