diff --git a/src/frontends/tensorflow_common/src/op/cumsum.cpp b/src/frontends/tensorflow_common/src/op/cumsum.cpp index f2b222fb70dc22..9e0343a2529a57 100644 --- a/src/frontends/tensorflow_common/src/op/cumsum.cpp +++ b/src/frontends/tensorflow_common/src/op/cumsum.cpp @@ -32,7 +32,7 @@ OutputVector translate_cumsum_op(const NodeContext& node) { auto const_one = create_same_type_const_scalar(axis, 1); auto axis_update = make_shared(less_than_zero, const_one, zero); - auto new_axis = make_shared(axis, axis_update); + axis = make_shared(axis, axis_update)->output(0); } auto cum_sum = make_shared(x, axis, exclusive, reverse);