diff --git a/datanway.py b/datanway.py
index 55a4c39..189cf63 100644
--- a/datanway.py
+++ b/datanway.py
@@ -85,6 +85,6 @@ def packslice(self, data_pack, numsamples):
 
             support_cacheX.append(slice_x)
             support_cacheY.append(keras.utils.to_categorical(slice_y,self.classes_per_set))
-            target_cacheY.append(keras.utils.to_categorical(target_y,self.classes_per_set)[0])
+            target_cacheY.append(keras.utils.to_categorical(target_y,self.classes_per_set))
             
-        return np.array(support_cacheX), np.array(support_cacheY), np.array(target_cacheY)
\ No newline at end of file
+        return np.array(support_cacheX), np.array(support_cacheY), np.array(target_cacheY)