diff --git a/src/layers/batch_norm.cu b/src/layers/batch_norm.cu
index 5e6c9b6..cea7477 100644
--- a/src/layers/batch_norm.cu
+++ b/src/layers/batch_norm.cu
@@ -190,5 +190,7 @@ float *BatchNorm2d::forward(const float *d_input) {
         CUDA_CHECK(cudaGetLastError());
     }
 
+    activation->activate(d_output);
+
     return d_output;
 }
\ No newline at end of file