1 files changed, 2 insertions, 2 deletions

diff --git a/src/bin/mnist/src/main.c b/src/bin/mnist/src/main.c
index 4d268ac..9aa3ce5 100644
--- a/src/bin/mnist/src/main.c
+++ b/src/bin/mnist/src/main.c
@@ -252,8 +252,8 @@ static void OneHotEncode(const uint8_t* labels_bytes, int num_labels, nnMatrix*
 
 static int OneHotDecode(const nnMatrix* label_matrix) {
   assert(label_matrix);
-  assert(label_matrix->cols == 1);
-  assert(label_matrix->rows == 10);
+  assert(label_matrix->cols == 10);
+  assert(label_matrix->rows == 1);
 
   R max_value = 0;
   int pos_max = 0;