diff --git a/include/aidge/operator/Memorize.hpp b/include/aidge/operator/Memorize.hpp
index 53340f8662f1fe1f9ca8795edb5a1e0fecc1cfa7..8991ccb44eb4926f375ff102858f4683e1bea4d8 100644
--- a/include/aidge/operator/Memorize.hpp
+++ b/include/aidge/operator/Memorize.hpp
@@ -38,7 +38,7 @@ public:
     using attr = typename Attributes_::template attr<e>;
 
     Memorize_Op(const unsigned int endStep)
-        : OperatorTensor(Type, 2, 0, 2),
+        : OperatorTensor(Type, 1, 1, 2),
           Attributes_(attr<MemorizeAttr::ScheduleStep>(0),
                       attr<MemorizeAttr::ForwardStep>(0),
                       attr<MemorizeAttr::EndStep>(endStep))
diff --git a/unit_tests/operator/Test_MetaOperator.cpp b/unit_tests/operator/Test_MetaOperator.cpp
index 328492d43be78ba3260ab914c1424066a137379e..421dec584c6edbbf11f229741ed85c1605474c8c 100644
--- a/unit_tests/operator/Test_MetaOperator.cpp
+++ b/unit_tests/operator/Test_MetaOperator.cpp
@@ -60,7 +60,7 @@ TEST_CASE("[core/operators] MetaOperator", "[Operator]") {
         microGraph->save("lstm", false, false);
 
         REQUIRE(myLSTM->nbInputs() == 3 + 8 + 8);
-        REQUIRE(myLSTM->nbData() == 3);
+        REQUIRE(myLSTM->nbData() == 1);
         REQUIRE(myLSTM->nbOutputs() == 2);
 
         std::shared_ptr<Tensor> myInput = std::make_shared<Tensor>();