1 files changed, 2 insertions, 4 deletions
diff --git a/src/main/java/com/google/devtools/build/lib/skyframe/SkylarkImportUniqueCycleFunction.java b/src/main/java/com/google/devtools/build/lib/skyframe/SkylarkImportUniqueCycleFunction.java
index e7c4decae7..a16ad9ab0a 100644
--- a/src/main/java/com/google/devtools/build/lib/skyframe/SkylarkImportUniqueCycleFunction.java
+++ b/src/main/java/com/google/devtools/build/lib/skyframe/SkylarkImportUniqueCycleFunction.java
@@ -14,14 +14,13 @@
 package com.google.devtools.build.lib.skyframe;
 
 import com.google.devtools.build.lib.cmdline.PackageIdentifier;
-import com.google.devtools.build.skyframe.SkyKey;
 import com.google.devtools.build.skyframe.SkyValue;
 
 /**
  * Emits an error message exactly once when a Skylark import cycle is found when running inlined
  * {@link SkylarkImportLookupFunction}s.
  */
-class SkylarkImportUniqueCycleFunction extends AbstractChainUniquenessFunction<SkyKey> {
+class SkylarkImportUniqueCycleFunction extends AbstractChainUniquenessFunction<PackageIdentifier> {
   private static final SkyValue INSTANCE = new SkyValue() {};
 
   @Override
@@ -45,8 +44,7 @@ class SkylarkImportUniqueCycleFunction extends AbstractChainUniquenessFunction<S
   }
 
   @Override
-  protected String elementToString(SkyKey elt) {
-    PackageIdentifier pkgId = (PackageIdentifier) elt.argument();
+  protected String elementToString(PackageIdentifier pkgId) {
     return pkgId.getPathFragment().getPathString();
   }
 }