diff --git a/src/layers/avg_pooling.cu b/src/layers/avg_pooling.cu
index aedcc13..b37bb9b 100644
--- a/src/layers/avg_pooling.cu
+++ b/src/layers/avg_pooling.cu
@@ -43,5 +43,7 @@ float* AvgPooling2D::forward(const float* d_input) {
         d_input, d_output, inputSize, outputSize, nChannels, poolingSize, stride
     );
 
+    activation.activate(d_output);
+
     return d_output;
 }
\ No newline at end of file
diff --git a/src/layers/max_pooling.cu b/src/layers/max_pooling.cu
index 3f1f426..f6ae351 100644
--- a/src/layers/max_pooling.cu
+++ b/src/layers/max_pooling.cu
@@ -46,5 +46,7 @@ float* MaxPooling2D::forward(const float* d_input) {
         d_input, d_output, inputSize, outputSize, nChannels, poolingSize, stride
     );
 
+    activation.activate(d_output);
+
     return d_output;
 }
\ No newline at end of file