diff --git a/test/GenericLayerTests.cpp b/test/GenericLayerTests.cpp
index 961ab16..188c7b1 100644
--- a/test/GenericLayerTests.cpp
+++ b/test/GenericLayerTests.cpp
@@ -87,9 +87,19 @@
     // Test Broadcast on add/mul operators
     HalPolicy::Model model2 = {};
 
-    AddInputOperand<HalPolicy>(model2, hidl_vec<uint32_t>{1, 1, 3, 4});
-    AddInputOperand<HalPolicy>(model2, hidl_vec<uint32_t>{4});
-    AddIntOperand<HalPolicy>(model2, actValue);
+    AddInputOperand<HalPolicy>(model2,
+                               hidl_vec<uint32_t>{1, 1, 3, 4},
+                               HalPolicy::OperandType::TENSOR_FLOAT32,
+                               0.0f,
+                               0,
+                               2);
+    AddInputOperand<HalPolicy>(model2,
+                               hidl_vec<uint32_t>{4},
+                               HalPolicy::OperandType::TENSOR_FLOAT32,
+                               0.0f,
+                               0,
+                               2);
+    AddIntOperand<HalPolicy>(model2, actValue, 2);
     AddOutputOperand<HalPolicy>(model2, hidl_vec<uint32_t>{1, 1, 3, 4});
     AddOutputOperand<HalPolicy>(model2, hidl_vec<uint32_t>{1, 1, 3, 4});
 
@@ -181,8 +191,16 @@
                                HalPolicy::OperandType::TENSOR_INT32);
     AddInputOperand<HalPolicy>(model,
                                hidl_vec<uint32_t>{4},
-                               HalPolicy::OperandType::TENSOR_INT32);
-    AddInputOperand<HalPolicy>(model, hidl_vec<uint32_t>{1, 1, 3, 4});
+                               HalPolicy::OperandType::TENSOR_INT32,
+                               0.0f,
+                               0,
+                               2);
+    AddInputOperand<HalPolicy>(model,
+                               hidl_vec<uint32_t>{1, 1, 3, 4},
+                               HalPolicy::OperandType::TENSOR_FLOAT32,
+                               0.0f,
+                               0,
+                               2);
 
     AddOutputOperand<HalPolicy>(model, hidl_vec<uint32_t>{1, 1, 3, 4});
     AddOutputOperand<HalPolicy>(model,