diff --git a/src/backends/reference/workloads/BatchMatMulImpl.cpp b/src/backends/reference/workloads/BatchMatMulImpl.cpp
index 74a358c..6693f15 100644
--- a/src/backends/reference/workloads/BatchMatMulImpl.cpp
+++ b/src/backends/reference/workloads/BatchMatMulImpl.cpp
@@ -73,7 +73,8 @@
 void BatchMatMul::AdjustAxesToMulForUnequalRanks(
     std::pair<std::pair<unsigned int, unsigned int>, std::pair<unsigned int, unsigned int>>& axesToMul)
 {
-    long rankDiff = static_cast<long>(inputXInfo.GetNumDimensions()) - inputYInfo.GetNumDimensions();
+    int rankDiff = static_cast<int>(inputXInfo.GetNumDimensions()) -
+                   static_cast<int>(inputYInfo.GetNumDimensions());
     if(rankDiff == 0)
     {
         return;