diff --git a/arm_compute/graph/Types.h b/arm_compute/graph/Types.h
index 190258b..cc3e346 100644
--- a/arm_compute/graph/Types.h
+++ b/arm_compute/graph/Types.h
@@ -149,6 +149,7 @@
     ResizeLayer,
     SoftmaxLayer,
     SplitLayer,
+    YOLOLayer,
 
     Input,
     Output,