diff --git a/aidge_export_cpp/static/activation_utils.hpp b/aidge_export_cpp/static/activation_utils.hpp
index c6a1bcdc0ce289a384519673ca04a001a5ca9692..ddac7fb46061c5ca909d1f0571723f0842dbda64 100644
--- a/aidge_export_cpp/static/activation_utils.hpp
+++ b/aidge_export_cpp/static/activation_utils.hpp
@@ -48,9 +48,5 @@ Output_T activation_forward_value (Sum_T weightedSum,
             break;
     }
 
-    // Value fixed here for now but it should be generated by
-    // the export module or determined by the type of Output_T
-    // For now only works for int8_t and uint8_t
-    const uint32_t NB_BITS = 8;
-    return saturate<Output_T>(rescaling(weightedSum, output), NB_BITS);
+    return saturate<Output_T>(rescaling(weightedSum, output), 8 * sizeof(Output_T));
 }