diff --git a/src/tim/vx/ops/roi_align_test.cc b/src/tim/vx/ops/roi_align_test.cc
index bd21542..ad45608 100644
--- a/src/tim/vx/ops/roi_align_test.cc
+++ b/src/tim/vx/ops/roi_align_test.cc
@@ -102,5 +102,5 @@ TEST(RoiAlign, shape_4_2_1_1_float32) {
 
   std::vector<float> output(num_rois * out_height * out_width * depth);
   EXPECT_TRUE(output_tensor->CopyDataFromTensor(output.data()));
-  EXPECT_EQ(golden, output);
+  ArraysMatch(golden, output, 1e-5f);
 }
\ No newline at end of file