diff --git a/unit_tests/operator/Test_MaxPoolingImpl.cpp b/unit_tests/operator/Test_MaxPoolingImpl.cpp
index f954ab2c2437ea0ab20e458a60a5be464e6b5cef..9f528f2d044cf43133f3729a7f0e4f1bd95b8889 100644
--- a/unit_tests/operator/Test_MaxPoolingImpl.cpp
+++ b/unit_tests/operator/Test_MaxPoolingImpl.cpp
@@ -77,7 +77,7 @@ TEST_CASE("[cpu/operator] MaxPooling(forward)", "[MaxPooling][CPU]") {
         myMaxPool->getOperator()->setBackend("cpu");
         op->computeOutputDims();
         myMaxPool->forward();
-        op->output(0).print();
+        op->getOutput(0)->print();
         REQUIRE(*(op->getOutput(0)) == *myOutput);
     }
 }
\ No newline at end of file
diff --git a/unit_tests/operator/Test_SliceImpl.cpp b/unit_tests/operator/Test_SliceImpl.cpp
index c291268a1d48661e6e2b05106dbc5576bc5549d5..3e25c28f9caac61c64d38fa70879af79d20392bc 100644
--- a/unit_tests/operator/Test_SliceImpl.cpp
+++ b/unit_tests/operator/Test_SliceImpl.cpp
@@ -36,8 +36,8 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice][CPU]") {
         mySlice->forward();
         // mySlice->getOperator()->output(0).print();
         REQUIRE(*(op->getOutput(0)) == *expectedOutput);
-        REQUIRE(op->output(0).dims() == expectedOutput->dims());
-        REQUIRE(op->output(0).dataType() == expectedOutput->dataType());
+        REQUIRE(op->getOutput(0)->dims() == expectedOutput->dims());
+        REQUIRE(op->getOutput(0)->dataType() == expectedOutput->dataType());
     }
 
     SECTION("2D Tensor") {
@@ -63,8 +63,8 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice][CPU]") {
         mySlice->forward();
         // mySlice->getOperator()->output(0).print();
         REQUIRE(*(op->getOutput(0)) == *expectedOutput);
-        REQUIRE(op->output(0).dims() == expectedOutput->dims());
-        REQUIRE(op->output(0).dataType() == expectedOutput->dataType());
+        REQUIRE(op->getOutput(0)->dims() == expectedOutput->dims());
+        REQUIRE(op->getOutput(0)->dataType() == expectedOutput->dataType());
     }
 
     SECTION("3D Tensor") {
@@ -97,8 +97,8 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice][CPU]") {
         mySlice->forward();
         // mySlice->getOperator()->output(0).print();
         REQUIRE(*(op->getOutput(0)) == *expectedOutput);
-        REQUIRE(op->output(0).dims() == expectedOutput->dims());
-        REQUIRE(op->output(0).dataType() == expectedOutput->dataType());
+        REQUIRE(op->getOutput(0)->dims() == expectedOutput->dims());
+        REQUIRE(op->getOutput(0)->dataType() == expectedOutput->dataType());
     }
 
     SECTION("4D Tensor") {
@@ -160,7 +160,7 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice][CPU]") {
         mySlice->forward();
         // mySlice->getOperator()->output(0).print();
         REQUIRE(*(op->getOutput(0)) == *expectedOutput);
-        REQUIRE(op->output(0).dims() == expectedOutput->dims());
-        REQUIRE(op->output(0).dataType() == expectedOutput->dataType());
+        REQUIRE(op->getOutput(0)->dims() == expectedOutput->dims());
+        REQUIRE(op->getOutput(0)->dataType() == expectedOutput->dataType());
     }
 }
\ No newline at end of file