2 files changed, 2 insertions, 2 deletions
diff --git a/augment.py b/augment.py
index c0f8c7a..e330328 100755
--- a/augment.py
+++ b/augment.py
@@ -100,7 +100,7 @@ if not opt.engine and not opt.onnx:
 sequence = read_sequence(data_opt.sequence_name, '')
 print("Got sequence {}, {} images".format(data_opt.sequence_name, len(sequence)))
 if data_opt.no_symlinks:
-  _len = len(sequence) - data_opt.augment_take - 2
+  _len = len(sequence) - data_opt.augment_make - 2
 else:
   _len = len(sequence)
 
diff --git a/build_dataset.pl b/build_dataset.pl
index 8b80d84..9d77766 100755
--- a/build_dataset.pl
+++ b/build_dataset.pl
@@ -4,7 +4,7 @@ use strict;
 use Cwd qw(cwd);
 my $base_dir = cwd;
 
-our $tag = lc shift @ARGV;
+our $tag = shift @ARGV;
 
 our $images_dir = $base_dir . "/sequences/$tag/";
 our $thumbs_dir = $base_dir . "/datasets/$tag/";