diff --git a/tests/src/core/memory.cpp b/tests/src/core/memory.cpp
index 63cc509a2..066de1a20 100644
--- a/tests/src/core/memory.cpp
+++ b/tests/src/core/memory.cpp
@@ -169,5 +169,5 @@ void testCast() {
   ASSERT_TRUE(occa::dtype::double_ == occa_memory.dtype());
   ASSERT_TRUE(occa::dtype::byte == casted_memory.dtype());
 
-  ASSERT_EQ(occa_memory.size(), casted_memory.size());
+  ASSERT_EQ(occa_memory.byte_size(), casted_memory.byte_size());
 }