diff --git a/library/core/src/main/java/com/google/android/exoplayer2/upstream/cache/CacheKeyFactory.java b/library/core/src/main/java/com/google/android/exoplayer2/upstream/cache/CacheKeyFactory.java
index 1e173ca3ad..bfa404c074 100644
--- a/library/core/src/main/java/com/google/android/exoplayer2/upstream/cache/CacheKeyFactory.java
+++ b/library/core/src/main/java/com/google/android/exoplayer2/upstream/cache/CacheKeyFactory.java
@@ -15,16 +15,15 @@
  */
 package com.google.android.exoplayer2.upstream.cache;
 
-import android.net.Uri;
 import com.google.android.exoplayer2.upstream.DataSpec;
 
-/** Factory for custom cache keys. */
+/** Factory for cache keys. */
 public interface CacheKeyFactory {
 
   /**
-   * Returns a custom cache key for the given {@link Uri}.
+   * Returns a cache key for the given {@link DataSpec}.
    *
-   * @param dataSpec The upstream source from which data will be read and cached.
+   * @param dataSpec The data being cached.
    */
   String buildCacheKey(DataSpec dataSpec);
 }