diff --git a/extensions/ima/src/main/java/com/google/android/exoplayer2/ext/ima/ImaServerSideAdInsertionMediaSource.java b/extensions/ima/src/main/java/com/google/android/exoplayer2/ext/ima/ImaServerSideAdInsertionMediaSource.java
index 27ea2275ce1..d97e107c373 100644
--- a/extensions/ima/src/main/java/com/google/android/exoplayer2/ext/ima/ImaServerSideAdInsertionMediaSource.java
+++ b/extensions/ima/src/main/java/com/google/android/exoplayer2/ext/ima/ImaServerSideAdInsertionMediaSource.java
@@ -374,8 +374,7 @@ private static State fromBundle(Bundle bundle) {
 
     private final ServerSideAdInsertionConfiguration configuration;
     private final Context context;
-    private final Map<ImaServerSideAdInsertionMediaSource, MediaSourceResourceHolder>
-        mediaSourceResources;
+    private final Map<String, MediaSourceResourceHolder> mediaSourceResources;
     private final Map<String, AdPlaybackState> adPlaybackStateMap;
 
     @Nullable private Player player;
@@ -401,6 +400,35 @@ public void setPlayer(Player player) {
       this.player = player;
     }
 
+    /**
+     * Puts the focus on the skip button, if a skip button is present and an ad is playing.
+     *
+     * @see StreamManager#focus()
+     */
+    public void focusSkipButton() {
+      if (player == null) {
+        return;
+      }
+      if (player.getPlaybackState() != Player.STATE_IDLE
+          && player.getPlaybackState() != Player.STATE_ENDED
+          && player.getMediaItemCount() > 0) {
+        int currentPeriodIndex = player.getCurrentPeriodIndex();
+        Object adsId =
+            player
+                .getCurrentTimeline()
+                .getPeriod(currentPeriodIndex, new Timeline.Period())
+                .getAdsId();
+        if (adsId instanceof String) {
+          MediaSourceResourceHolder mediaSourceResourceHolder = mediaSourceResources.get(adsId);
+          if (mediaSourceResourceHolder != null
+              && mediaSourceResourceHolder.imaServerSideAdInsertionMediaSource.streamManager
+                  != null) {
+            mediaSourceResourceHolder.imaServerSideAdInsertionMediaSource.streamManager.focus();
+          }
+        }
+      }
+    }
+
     /**
      * Releases resources.
      *
@@ -427,7 +455,7 @@ private void addMediaSourceResources(
         StreamPlayer streamPlayer,
         com.google.ads.interactivemedia.v3.api.AdsLoader adsLoader) {
       mediaSourceResources.put(
-          mediaSource, new MediaSourceResourceHolder(mediaSource, streamPlayer, adsLoader));
+          mediaSource.adsId, new MediaSourceResourceHolder(mediaSource, streamPlayer, adsLoader));
     }
 
     private AdPlaybackState getAdPlaybackState(String adsId) {