diff --git a/dalvik/src/test/java/com/ibm/wala/dalvik/test/callGraph/JVMLDalvikComparisonTest.java b/dalvik/src/test/java/com/ibm/wala/dalvik/test/callGraph/JVMLDalvikComparisonTest.java
index 70877ad97b..9a993b4d3f 100644
--- a/dalvik/src/test/java/com/ibm/wala/dalvik/test/callGraph/JVMLDalvikComparisonTest.java
+++ b/dalvik/src/test/java/com/ibm/wala/dalvik/test/callGraph/JVMLDalvikComparisonTest.java
@@ -95,23 +95,14 @@ private static Set<Pair<CGNode, CGNode>> edgeDiff(
     return result;
   }
 
-  private static void test(String mainClass, String javaScopeFile)
-      throws IllegalArgumentException, IOException, CancelException, ClassHierarchyException {
-    test(mainClass, javaScopeFile, false);
-  }
-
   /**
    * Run tests to compare the call graphs computing with the JVM bytecode frontend vs the Dalvik
    * frontend
    *
    * @param mainClass main class for the test
    * @param javaScopeFile scope file for the test
-   * @param allowExtraJavaCGEdges if true, allow extra edges in the JVM bytecode frontend call
-   *     graph. This flag is temporarily required due to issues with JLex caused by
-   *     https://issuetracker.google.com/issues/316744331. TODO: remove this flag once the D8 issue
-   *     is fixed.
    */
-  private static void test(String mainClass, String javaScopeFile, boolean allowExtraJavaCGEdges)
+  private static void test(String mainClass, String javaScopeFile)
       throws IllegalArgumentException, IOException, CancelException, ClassHierarchyException {
     Pair<CallGraph, PointerAnalysis<InstanceKey>> java = makeJavaBuilder(javaScopeFile, mainClass);
 
@@ -124,11 +115,9 @@ private static void test(String mainClass, String javaScopeFile, boolean allowEx
     Set<MethodReference> androidMethods = applicationMethods(android.fst);
     Set<MethodReference> javaMethods = applicationMethods(java.fst);
 
-    if (!allowExtraJavaCGEdges) {
-      Set<Pair<CGNode, CGNode>> javaExtraEdges = edgeDiff(java.fst, android.fst, false);
-      assert !checkEdgeDiff(android, androidMethods, javaMethods, javaExtraEdges)
-          : "found extra edges in Java call graph";
-    }
+    Set<Pair<CGNode, CGNode>> javaExtraEdges = edgeDiff(java.fst, android.fst, false);
+    assert !checkEdgeDiff(android, androidMethods, javaMethods, javaExtraEdges)
+        : "found extra edges in Java call graph";
 
     Set<Pair<CGNode, CGNode>> androidExtraEdges = edgeDiff(android.fst, java.fst, true);
     assert !checkEdgeDiff(java, javaMethods, androidMethods, androidExtraEdges)
@@ -221,7 +210,7 @@ private static boolean checkEdgeDiff(
   @Test
   public void testJLex()
       throws ClassHierarchyException, IllegalArgumentException, IOException, CancelException {
-    test(TestConstants.JLEX_MAIN, TestConstants.JLEX, true);
+    test(TestConstants.JLEX_MAIN, TestConstants.JLEX);
   }
 
   @Test
diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml
index 5766673be4..48a7526cbf 100644
--- a/gradle/libs.versions.toml
+++ b/gradle/libs.versions.toml
@@ -6,7 +6,7 @@ ktfmt = "0.44"
 spotless = "6.23.3"
 
 [libraries]
-android-tools = "com.android.tools:r8:8.2.39"
+android-tools = "com.android.tools:r8:8.2.46"
 ant = "org.apache.ant:ant:1.10.14"
 assertj-core = "org.assertj:assertj-core:3.24.2"
 commons-cli = "commons-cli:commons-cli:1.5.0"