diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 9f7d544..5cb4b6a 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -235,7 +235,7 @@
     SparseToDense = OperatorInfo()
     Split = OperatorInfo(indices=SPLIT_IFM_INDICES)
     SplitSliceRead = OperatorInfo(indices=IFM_INDICES)
-    SplitV = OperatorInfo(indices=IFM_IFM2_INDICES)
+    SplitV = OperatorInfo(indices=IFM_INDICES)
     Sqrt = OperatorInfo()
     Square = OperatorInfo()
     SquaredDifference = OperatorInfo()
diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py
index f7dfec2..a1fcf6a 100644
--- a/ethosu/vela/supported_operators.py
+++ b/ethosu/vela/supported_operators.py
@@ -708,7 +708,7 @@
     @staticmethod
     def constraint_splitv_inferred(op):
         "Only one size is allowed to be inferred"
-        sizes = op.ifm2.values
+        sizes = op.inputs[1].values
         valid = np.count_nonzero(sizes == -1) <= 1
         return valid, f"Op has multiple inferred sizes (-1): {sizes}"