diff --git a/fog_x/rlds/utils.py b/fog_x/rlds/utils.py
index 35e6d62..11ad695 100644
--- a/fog_x/rlds/utils.py
+++ b/fog_x/rlds/utils.py
@@ -1,6 +1,5 @@
 import numpy as np
 import tensorflow_datasets as tfds  # type: ignore
-from IPython import display
 from PIL import Image
 
 DATASETS = [