diff --git a/ethosu/regor/tosa/tosa_reader.cpp b/ethosu/regor/tosa/tosa_reader.cpp
index 458955f73f36194080fa7e26a032b6aa970b5335..1cb7db619093de282d89caeb45e983dd9278c7f8 100644
--- a/ethosu/regor/tosa/tosa_reader.cpp
+++ b/ethosu/regor/tosa/tosa_reader.cpp
@@ -753,6 +753,13 @@ void TosaReader::LoadGraphs(const tosaFb::TosaGraph *model, std::list<GraphBuild
                             "Failed to set OUTPAD attribute on TRANSPOSE_CONV2D");
                     }
                     break;
+                    case tosaFb::Op::MUL:
+                    {
+                        const auto &mulShiftTensor = input_tensors[2];
+                        tosa_assert(shapes.at(mulShiftTensor).count == 1,
+                            ("MUL shift tensor " + mulShiftTensor + " needs to have rank 1.").c_str());
+                    }
+                    break;
                     default:
                         break;
                 }