diff --git a/facebook/src/test/java/com/facebook/appevents/AppEventsLoggerTest.java b/facebook/src/test/java/com/facebook/appevents/AppEventsLoggerTest.java
index 670b368dda..6f465f1bcf 100644
--- a/facebook/src/test/java/com/facebook/appevents/AppEventsLoggerTest.java
+++ b/facebook/src/test/java/com/facebook/appevents/AppEventsLoggerTest.java
@@ -72,9 +72,8 @@
         AppEventQueue.class,
         AppEventUtility.class,
         AppEventsLogger.class,
-        AppEventsLoggerImpl.class,
+        AppEventsLogger.class,
         AttributionIdentifiers.class,
-        InternalAppEventsLogger.class,
         FacebookSdk.class,
         GraphRequest.class,
         Utility.class,
@@ -281,6 +280,7 @@ public void testLogProductItemWithoutGtinMpnBrand() throws Exception {
                 Matchers.anyDouble(),
                 argThat(new AppEventTestUtilities.BundleMatcher(parameters)),
                 Matchers.anyBoolean(),
+                Matchers.anyBoolean(),
                 Matchers.any(UUID.class));
     }
 
@@ -298,6 +298,7 @@ public void testLogPushNotificationOpen() throws Exception {
                 Matchers.anyDouble(),
                 argThat(new AppEventTestUtilities.BundleMatcher(parameters)),
                 Matchers.anyBoolean(),
+                Matchers.anyBoolean(),
                 Matchers.any(UUID.class));
     }
 
@@ -313,6 +314,7 @@ public void testLogPushNotificationOpenWithoutCampaign() throws Exception {
                 Matchers.anyDouble(),
                 Matchers.any(Bundle.class),
                 Matchers.anyBoolean(),
+                Matchers.anyBoolean(),
                 Matchers.any(UUID.class));
     }
 
@@ -347,6 +349,7 @@ public void testLogPushNotificationOpenWithoutPayload() throws Exception {
                 Matchers.anyDouble(),
                 Matchers.any(Bundle.class),
                 Matchers.anyBoolean(),
+                Matchers.anyBoolean(),
                 Matchers.any(UUID.class));
     }