diff --git a/library/core/src/test/java/com/google/android/exoplayer2/drm/OfflineLicenseHelperTest.java b/library/core/src/test/java/com/google/android/exoplayer2/drm/OfflineLicenseHelperTest.java
index f7b249765b..d57b3bb6e8 100644
--- a/library/core/src/test/java/com/google/android/exoplayer2/drm/OfflineLicenseHelperTest.java
+++ b/library/core/src/test/java/com/google/android/exoplayer2/drm/OfflineLicenseHelperTest.java
@@ -52,10 +52,10 @@ public class OfflineLicenseHelperTest {
             new ExoMediaDrm.KeyRequest(/* data= */ new byte[0], /* licenseServerUrl= */ ""));
     offlineLicenseHelper =
         new OfflineLicenseHelper(
-            C.WIDEVINE_UUID,
-            new ExoMediaDrm.AppManagedProvider(mediaDrm),
-            mediaDrmCallback,
-            /* optionalKeyRequestParameters= */ null,
+            new DefaultDrmSessionManager.Builder()
+                .setUuidAndExoMediaDrmProvider(
+                    C.WIDEVINE_UUID, new ExoMediaDrm.AppManagedProvider(mediaDrm))
+                .build(mediaDrmCallback),
             new DrmSessionEventListener.EventDispatcher());
   }