diff --git a/examples/1. simple_image_classifier/test1.npz b/examples/1. simple_image_classifier/test1.npz
index 9f26f59..7185b45 100644
Binary files a/examples/1. simple_image_classifier/test1.npz and b/examples/1. simple_image_classifier/test1.npz differ
diff --git a/examples/2. mnist/mnist.npz b/examples/2. mnist/mnist.npz
index b9c59da..fae3ccf 100644
Binary files a/examples/2. mnist/mnist.npz and b/examples/2. mnist/mnist.npz differ
diff --git a/pod.py b/pod.py
index 229670e..4eed57b 100644
--- a/pod.py
+++ b/pod.py
@@ -51,6 +51,6 @@ def load(**options):
             if options.get("one_hot") != None:
                 Q =  []
                 for i in range(len(data)):
-                    Q.append(decimal_to_array(data[i], int(options.get("one-hot"))))
+                    Q.append(decimal_to_array(data[i], int(options.get("one_hot"))))
                 data = np.vstack(Q)
             return data
diff --git a/setup.py b/setup.py
index e7fd613..81c6457 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name='pileoffeather',
-    version='0.3.0',
+    version='0.3.1',
     license='MIT',
     url = 'https://github.com/usedToBeTomas/pile-of-feather',
     author='Daniele Tomaselli',