diff --git a/packages/SystemUI/tests/src/com/android/systemui/statusbar/notification/logging/NotificationMemoryMonitorTest.kt b/packages/SystemUI/tests/src/com/android/systemui/statusbar/notification/logging/NotificationMemoryMonitorTest.kt
index 1531f888c857a..16e2441c556bc 100644
--- a/packages/SystemUI/tests/src/com/android/systemui/statusbar/notification/logging/NotificationMemoryMonitorTest.kt
+++ b/packages/SystemUI/tests/src/com/android/systemui/statusbar/notification/logging/NotificationMemoryMonitorTest.kt
@@ -148,7 +148,7 @@ class NotificationMemoryMonitorTest : SysuiTestCase() {
             smallIcon = notification.smallIcon.bitmap.allocationByteCount,
             largeIcon = notification.getLargeIcon().bitmap.allocationByteCount,
             extras = 4092,
-            bigPicture = 960000,
+            bigPicture = bigPicture.bitmap.allocationByteCount,
             extender = 0,
             style = "BigPictureStyle",
             styleIcon = bigPictureIcon.bitmap.allocationByteCount,