diff --git a/get_train_val.py b/get_train_val.py
index 5955912678bd1a91aedeaac8c796b95c71792965..0666da9a6dc99b8fbadb3c6b8dec9a1b44a9d423 100755
--- a/get_train_val.py
+++ b/get_train_val.py
@@ -100,7 +100,7 @@ if __name__ == '__main__':
     train, val = utils.split(df, 'train', args.ratio)
     
     saved_directory = os.path.join(args.directory, 'set')
-
+    utils.create_directory(saved_directory)
     if args.test:
         val, test = utils.split(val, 'val', 0.5)
         export_split(args, [train, val, test], args.path_to_data,