diff --git a/sdk/diffgram/core/directory.py b/sdk/diffgram/core/directory.py
index 54d1a2a..ee42ba3 100644
--- a/sdk/diffgram/core/directory.py
+++ b/sdk/diffgram/core/directory.py
@@ -1,7 +1,6 @@
 from diffgram.file.file import File
 from ..regular.regular import refresh_from_dict
 import logging
-from diffgram.tensorflow_diffgram.diffgram_tensorflow_dataset import DiffgramTensorflowDataset
 from diffgram.core.diffgram_dataset_iterator import DiffgramDatasetIterator
 from multiprocessing.pool import ThreadPool as Pool
 
@@ -166,6 +165,7 @@ def to_pytorch(self, transform = None):
         return pytorch_dataset
 
     def to_tensorflow(self):
+        from diffgram.tensorflow_diffgram.diffgram_tensorflow_dataset import DiffgramTensorflowDataset
         file_id_list = self.file_id_list
         diffgram_tensorflow_dataset = DiffgramTensorflowDataset(
             project = self.client,
diff --git a/sdk/requirements.txt b/sdk/requirements.txt
index 02704a9..651c49d 100644
--- a/sdk/requirements.txt
+++ b/sdk/requirements.txt
@@ -5,4 +5,4 @@ six>=1.9.0
 tensorflow>=1.12.0
 pillow
 torch
-imageio
\ No newline at end of file
+imageio>=2.9.0
\ No newline at end of file
diff --git a/sdk/setup.py b/sdk/setup.py
index 9702a09..e7df02a 100644
--- a/sdk/setup.py
+++ b/sdk/setup.py
@@ -25,6 +25,7 @@
 	   'opencv-python>=4.0.0.21',
 	   'scipy>=1.1.0',
 	   'six>=1.9.0',
-	   'pillow>=6.1.0'
+	   'pillow>=6.1.0',
+       'imageio>=2.9.0'
 	]
 )
\ No newline at end of file