diff --git a/unit_tests/operator/Test_AddImpl.cpp b/unit_tests/operator/Test_AddImpl.cpp
index 579ded6cccc806418e0549b54eee521bbe046a4f..b36d61548b7bae84813fcd0012a85b23a51fc26e 100644
--- a/unit_tests/operator/Test_AddImpl.cpp
+++ b/unit_tests/operator/Test_AddImpl.cpp
@@ -47,7 +47,7 @@ TEST_CASE("[cpu/operator] Add(forward)", "[Add]") {
         myAdd->getOperator()->computeOutputDims();
         myAdd->forward();
 
-        REQUIRE(*std::static_pointer_cast<Tensor>(myAdd->getOperator()->getOutput(0)) == *input1);
+        REQUIRE(myAdd->getOperator()->output(0) == *input1);
     }
 
     SECTION("Two inputs") {