diff --git a/test/test_transaction.cpp b/test/test_transaction.cpp
index a81d3e07333fa76d4c4c888cb7bc984e198b8f3d..75a96eada22d77074ec4a0bbb62e1d5ba69c967e 100644
--- a/test/test_transaction.cpp
+++ b/test/test_transaction.cpp
@@ -79,8 +79,8 @@ TEST(test_transaction, retrieve_non_existing) {
   transaction->ExecuteAsynchronously();
 
   auto status = transaction->WaitForIt();
-  EXPECT_EQ(status.GetCode(), TransactionStatus::SUCCESS().GetCode());
-  ASSERT_EQ(status.GetCode(), StatusCode::SUCCESS);
+  EXPECT_EQ(status.GetCode(), TransactionStatus::TRANSACTION_ERROR().GetCode());
+  ASSERT_EQ(status.GetCode(), StatusCode::GENERIC_TRANSACTION_ERROR);
 
   const auto &result_set =
     dynamic_cast<const UniqueResult &>(transaction->GetResultSet());