diff --git a/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/input.h b/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/input.h
index c075960..ae75244 100644
--- a/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/input.h
+++ b/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/input.h
@@ -18,7 +18,7 @@
 
 /******************************************************************************************************************
  * Model: keyword_spotting_dnn_medium_int8
- * Source: https://github.com/ARM-software/ML-zoo/tree/master/models/keyword_spotting/dnn_medium/tflite_int8
+ * Source: https://github.com/ARM-software/ML-zoo/tree/22.02/models/keyword_spotting/dnn_medium/tflite_int8
  * Hexdump: python3 -c 'import numpy;i=numpy.load("input.npy");i.tofile("input.bin");' && xxd -i input.bin > input.h
  ******************************************************************************************************************/
 
@@ -36,4 +36,4 @@
     0x21, 0x90, 0xc5, 0x35, 0x6c, 0x3e, 0x7a, 0xf1, 0x53, 0xee, 0x90, 0x47, 0x4b, 0xf2, 0xfa, 0xa7, 0x76, 0x31,
     0x10, 0xbd, 0xf0, 0x04, 0x59, 0x45, 0x86, 0x0c, 0x6b, 0x72, 0xb5, 0x59, 0x0c, 0x05, 0x67, 0xfc, 0x8c, 0x80,
     0x82, 0x86, 0xbb, 0xbe, 0xbf, 0x5f, 0x80, 0x6e, 0x17, 0x2f, 0x6c, 0x5b, 0x7b, 0x04, 0xe3, 0x42, 0xbf, 0xf4,
-    0xac, 0x03, 0xdd, 0x18, 0x11, 0x61, 0xe8, 0x6f, 0x58, 0xc8, 0xa0, 0xef, 0x8e, 0x16, 0x2f, 0xc5};
\ No newline at end of file
+    0xac, 0x03, 0xdd, 0x18, 0x11, 0x61, 0xe8, 0x6f, 0x58, 0xc8, 0xa0, 0xef, 0x8e, 0x16, 0x2f, 0xc5};
diff --git a/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/model.h b/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/model.h
index f190887..24458ec 100644
--- a/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/model.h
+++ b/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/model.h
@@ -18,12 +18,14 @@
 
 /*************************************************************************************
  * Model: keyword_spotting_dnn_medium_int8
- * Source: https://github.com/ARM-software/ML-zoo/tree/master/models/keyword_spotting/dnn_medium/tflite_int8
+ * Source: https://github.com/ARM-software/ML-zoo/tree/22.02/models/keyword_spotting/dnn_medium/tflite_int8
  * Vela optimisation: vela --accelerator-config ethos-u55-32 dnn_m_quantized.tflite
  * Hexdump: xxd -i dnn_m_quantized_vela.tflite model.h
  ************************************************************************************/
 
-#define TENSOR_ARENA_SIZE 2000
+#include <stddef.h>
+
+const size_t tensorArenaSize = 2000;
 
 const char *modelName = "keyword_spotting_dnn_medium_int8";
 
@@ -10435,4 +10437,4 @@
     0x0e, 0x00, 0x00, 0x00, 0x56, 0x65, 0x6c, 0x61, 0x20, 0x4f, 0x70, 0x74, 0x69, 0x6d, 0x69, 0x73, 0x65, 0x64, 0x00,
     0x00, 0x01, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x10, 0x00, 0x0f, 0x00, 0x04, 0x00, 0x00, 0x00,
     0x08, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x20, 0x07,
-    0x00, 0x00, 0x00, 0x65, 0x74, 0x68, 0x6f, 0x73, 0x2d, 0x75, 0x00};
\ No newline at end of file
+    0x00, 0x00, 0x00, 0x65, 0x74, 0x68, 0x6f, 0x73, 0x2d, 0x75, 0x00};
diff --git a/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/output.h b/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/output.h
index 7458eff..f4c8a4b 100644
--- a/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/output.h
+++ b/applications/baremetal/models/ethos-u55-32/keyword_spotting_dnn_medium_int8/output.h
@@ -18,10 +18,10 @@
 
 /******************************************************************************************************************
  * Model: keyword_spotting_dnn_medium_int8
- * Source: https://github.com/ARM-software/ML-zoo/tree/master/models/keyword_spotting/dnn_medium/tflite_int8
+ * Source: https://github.com/ARM-software/ML-zoo/tree/22.02/models/keyword_spotting/dnn_medium/tflite_int8
  * Hexdump: python3 -c 'import numpy;i=numpy.load("output.npy");i.tofile("output.bin");' && xxd -i output.bin > output.h
  ******************************************************************************************************************/
 
 unsigned char expectedOutputData[] __attribute__((
     aligned(16),
-    section("expected_output_data_sec"))) = {0x80, 0x80, 0x80, 0x7f, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80};
\ No newline at end of file
+    section("expected_output_data_sec"))) = {0x80, 0x80, 0x80, 0x7f, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80};