diff --git a/include/aidge/operator/Expand.hpp b/include/aidge/operator/Expand.hpp
index da87cfb562efe650b434f73e486430da72ad0780..95ca72a2782959f036f09a7948c6fda46fbdf9f0 100644
--- a/include/aidge/operator/Expand.hpp
+++ b/include/aidge/operator/Expand.hpp
@@ -73,7 +73,7 @@ class Expand_Op
      */
     Expand_Op()
         : OperatorTensor(Type,
-                         {InputCategory::Data, InputCategory::Param},
+                         {InputCategory::Data, InputCategory::Data},
                          1) {}
 
     Expand_Op(const Expand_Op &op);