diff --git a/reference_model/src/ops/data_layout.cc b/reference_model/src/ops/data_layout.cc
index f3e80f3..674227b 100644
--- a/reference_model/src/ops/data_layout.cc
+++ b/reference_model/src/ops/data_layout.cc
@@ -89,7 +89,7 @@
         }
     }
 
-    ERROR_IF(output_dim_on_axis == outputs[0]->getShape()[attribute->axis()],
+    ERROR_IF(output_dim_on_axis != outputs[0]->getShape()[attribute->axis()],
              "OpConcat: sum of input dimension on axis not equal to output dimension on axis");
 
     out = dynamic_cast<TosaReference::TensorTemplate<TOut>*>(outputs[0]);