diff --git a/src/utils/__tests__/notification.test.js b/src/utils/__tests__/notification.test.js
index ca4eb5f1..d6626311 100644
--- a/src/utils/__tests__/notification.test.js
+++ b/src/utils/__tests__/notification.test.js
@@ -12,7 +12,7 @@ describe('device utils', () => {
         ];
 
         types.forEach((type) => {
-            expect(nUtils.getColor(type)).toMatchSnapshot();
+            expect(nUtils.getPrimaryColor(type)).toMatchSnapshot();
         });
     });
     it('get icon', () => {