diff --git a/nvdsinfer_custom_impl_Yolo/layers/activation_layer.h b/nvdsinfer_custom_impl_Yolo/layers/activation_layer.h
index e6081e6..176a5c2 100644
--- a/nvdsinfer_custom_impl_Yolo/layers/activation_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/activation_layer.h
@@ -8,6 +8,7 @@
 
 #include <string>
 #include <cassert>
+#include <iostream>
 
 #include "NvInfer.h"