diff --git a/android/src/test/java/com/instabug/reactlibrary/RNInstabugReactnativeModuleTest.java b/android/src/test/java/com/instabug/reactlibrary/RNInstabugReactnativeModuleTest.java
index cb0e4912a..44bee2a15 100644
--- a/android/src/test/java/com/instabug/reactlibrary/RNInstabugReactnativeModuleTest.java
+++ b/android/src/test/java/com/instabug/reactlibrary/RNInstabugReactnativeModuleTest.java
@@ -638,20 +638,20 @@ public void testWillRedirectToStore() {
 
     @Test
     public void testEnableAutoMasking() {
-
+        // when
         rnModule.setAutoMaskingEnabled(true);
 
-        verify(Instabug.class, times(1));
-        Instabug.setNetworkAutoMaskingState(Feature.State.ENABLED);
+        // then
+        mockInstabug.verify(() -> Instabug.setNetworkAutoMaskingState(Feature.State.ENABLED));
     }
-    
+
     @Test
     public void testDisableAutoMasking() {
-
+        // when
         rnModule.setAutoMaskingEnabled(false);
 
-        verify(Instabug.class, times(1));
-        Instabug.setNetworkAutoMaskingState(Feature.State.DISABLED);
+        // then
+        mockInstabug.verify(() -> Instabug.setNetworkAutoMaskingState(Feature.State.DISABLED));
     }
 
 }