diff --git a/delegate/classic/src/ClassicDelegateUtils.hpp b/delegate/classic/src/ClassicDelegateUtils.hpp
index d378c6c..f5ff32a 100644
--- a/delegate/classic/src/ClassicDelegateUtils.hpp
+++ b/delegate/classic/src/ClassicDelegateUtils.hpp
@@ -437,7 +437,7 @@
         std::vector<unsigned char> dimensionsSpecificity(tensorDimensionSize, true);
         for (int i = 0; i < tensorDimensionSize; ++i) {
             auto dim = tfLiteTensor.dims->data[i];
-            if (dim <= 0)
+            if (dim < 0)
             {
                 dimensionsSpecificity[i] = false;
             }
diff --git a/delegate/opaque/src/OpaqueDelegateUtils.hpp b/delegate/opaque/src/OpaqueDelegateUtils.hpp
index a636161..792dc5d 100644
--- a/delegate/opaque/src/OpaqueDelegateUtils.hpp
+++ b/delegate/opaque/src/OpaqueDelegateUtils.hpp
@@ -509,7 +509,7 @@
         {
             int32_t dim = TfLiteOpaqueTensorDim(tfLiteTensor, i);
 
-            if (dim <= 0)
+            if (dim < 0)
             {
                 dimensionsSpecificity[i] = false;
             }