diff --git a/decloud/models/create_tfrecords.py b/decloud/models/create_tfrecords.py
index d9993007c7366aceff51d0a7c937cfcc7ba801f2..21b8ad547a8779f88580e891b3cd027fa7e8d706 100644
--- a/decloud/models/create_tfrecords.py
+++ b/decloud/models/create_tfrecords.py
@@ -22,7 +22,6 @@ 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