diff --git a/projects/hip-tests/samples/0_Intro/bit_extract/bit_extract.cpp b/projects/hip-tests/samples/0_Intro/bit_extract/bit_extract.cpp
index d43847946e..e87124f47f 100644
--- a/projects/hip-tests/samples/0_Intro/bit_extract/bit_extract.cpp
+++ b/projects/hip-tests/samples/0_Intro/bit_extract/bit_extract.cpp
@@ -63,9 +63,9 @@ int main(int argc, char* argv[]) {
 
     printf("info: allocate host mem (%6.2f MB)\n", 2 * Nbytes / 1024.0 / 1024.0);
     A_h = (uint32_t*)malloc(Nbytes);
-    CHECK(A_h == 0 ? hipErrorMemoryAllocation : hipSuccess);
+    CHECK(A_h == 0 ? hipErrorOutOfMemory : hipSuccess);
     C_h = (uint32_t*)malloc(Nbytes);
-    CHECK(C_h == 0 ? hipErrorMemoryAllocation : hipSuccess);
+    CHECK(C_h == 0 ? hipErrorOutOfMemory : hipSuccess);
 
     for (size_t i = 0; i < N; i++) {
         A_h[i] = i;
diff --git a/projects/hip-tests/samples/0_Intro/square/square.hipref.cpp b/projects/hip-tests/samples/0_Intro/square/square.hipref.cpp
index a99f5af9bd..6911b9f6c3 100644
--- a/projects/hip-tests/samples/0_Intro/square/square.hipref.cpp
+++ b/projects/hip-tests/samples/0_Intro/square/square.hipref.cpp
@@ -62,9 +62,9 @@ int main(int argc, char* argv[]) {
 #endif
     printf("info: allocate host mem (%6.2f MB)\n", 2 * Nbytes / 1024.0 / 1024.0);
     A_h = (float*)malloc(Nbytes);
-    CHECK(A_h == 0 ? hipErrorMemoryAllocation : hipSuccess);
+    CHECK(A_h == 0 ? hipErrorOutOfMemory : hipSuccess);
     C_h = (float*)malloc(Nbytes);
-    CHECK(C_h == 0 ? hipErrorMemoryAllocation : hipSuccess);
+    CHECK(C_h == 0 ? hipErrorOutOfMemory : hipSuccess);
     // Fill with Phi + i
     for (size_t i = 0; i < N; i++) {
         A_h[i] = 1.618f + i;