diff --git a/src/armnnTfParser/test/Split.cpp b/src/armnnTfParser/test/Split.cpp
index 87cd654..bf42bf7 100644
--- a/src/armnnTfParser/test/Split.cpp
+++ b/src/armnnTfParser/test/Split.cpp
@@ -106,7 +106,7 @@
         }
         m_Prototext += R"(
           attr {
-            key: "num_or_size_splits"
+            key: "num_split"
             value {
               i: 2
             }