diff --git a/libraries/exoplayer_hls/src/main/java/androidx/media3/exoplayer/hls/DefaultHlsDataSourceFactory.java b/libraries/exoplayer_hls/src/main/java/androidx/media3/exoplayer/hls/DefaultHlsDataSourceFactory.java
index d37f36bd8a..4ae3733351 100644
--- a/libraries/exoplayer_hls/src/main/java/androidx/media3/exoplayer/hls/DefaultHlsDataSourceFactory.java
+++ b/libraries/exoplayer_hls/src/main/java/androidx/media3/exoplayer/hls/DefaultHlsDataSourceFactory.java
@@ -15,6 +15,7 @@
  */
 package androidx.media3.exoplayer.hls;
 
+import androidx.media3.common.C;
 import androidx.media3.common.util.UnstableApi;
 import androidx.media3.datasource.DataSource;
 
@@ -32,7 +33,7 @@ public final class DefaultHlsDataSourceFactory implements HlsDataSourceFactory {
   }
 
   @Override
-  public DataSource createDataSource(int dataType) {
+  public DataSource createDataSource(@C.DataType int dataType) {
     return dataSourceFactory.createDataSource();
   }
 }
diff --git a/libraries/exoplayer_hls/src/main/java/androidx/media3/exoplayer/hls/HlsDataSourceFactory.java b/libraries/exoplayer_hls/src/main/java/androidx/media3/exoplayer/hls/HlsDataSourceFactory.java
index 716ec5cf6b..4ca31530c0 100644
--- a/libraries/exoplayer_hls/src/main/java/androidx/media3/exoplayer/hls/HlsDataSourceFactory.java
+++ b/libraries/exoplayer_hls/src/main/java/androidx/media3/exoplayer/hls/HlsDataSourceFactory.java
@@ -26,9 +26,8 @@ public interface HlsDataSourceFactory {
   /**
    * Creates a {@link DataSource} for the given data type.
    *
-   * @param dataType The data type for which the {@link DataSource} will be used. One of {@link C}
-   *     {@code .DATA_TYPE_*} constants.
+   * @param dataType The {@link C.DataType} for which the {@link DataSource} will be used.
    * @return A {@link DataSource} for the given data type.
    */
-  DataSource createDataSource(int dataType);
+  DataSource createDataSource(@C.DataType int dataType);
 }