diff --git a/projects/benchmark/classification/benchmark.py b/projects/benchmark/classification/benchmark.py
index 33d4b498..aa08ac4b 100644
--- a/projects/benchmark/classification/benchmark.py
+++ b/projects/benchmark/classification/benchmark.py
@@ -23,6 +23,7 @@ def param_count(model):
 
 
 def get_mean_std(mode="imagenet_default_mean_std"):
+    mode = mode.upper()
     if mode == "IMAGENET_DEFAULT_MEAN_STD":
         mean = (0.485, 0.456, 0.406)
         std = (0.229, 0.224, 0.225)
diff --git a/setup.py b/setup.py
index 554ef323..44952cf4 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,7 @@ def get_dist(pkgname):
         return None
 
 
-version = "0.0.6"
+version = "0.1.0"
 package_name = "flowvision"
 cwd = os.path.dirname(os.path.abspath(__file__))