diff --git a/decloud/models/create_tfrecords.py b/decloud/models/create_tfrecords.py
index 21b8ad547a8779f88580e891b3cd027fa7e8d706..d9993007c7366aceff51d0a7c937cfcc7ba801f2 100644
--- a/decloud/models/create_tfrecords.py
+++ b/decloud/models/create_tfrecords.py
@@ -22,6 +22,7 @@ FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 DEALINGS IN THE SOFTWARE.
 """
 """Create some TFRecords from a decloud.dataset"""
+import os
 import argparse
 import sys
 import logging
diff --git a/decloud/preprocessing/sentinel2_prepare.py b/decloud/preprocessing/sentinel2_prepare.py
index 9e7221eca1bfb1a5783158e956a3dd0a6cd08ead..987ccecac69f77e93b27e1b3c91a50dadb9f1df9 100644
--- a/decloud/preprocessing/sentinel2_prepare.py
+++ b/decloud/preprocessing/sentinel2_prepare.py
@@ -22,6 +22,7 @@ FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 DEALINGS IN THE SOFTWARE.
 """
 """Pre-process one Sentinel-2 image"""
+import os
 import logging
 import argparse
 import otbApplication