diff --git a/extensions/media2/src/main/java/com/google/android/exoplayer2/ext/media2/PlayerWrapper.java b/extensions/media2/src/main/java/com/google/android/exoplayer2/ext/media2/PlayerWrapper.java
index dbf0f68f63..b04ff5f3de 100644
--- a/extensions/media2/src/main/java/com/google/android/exoplayer2/ext/media2/PlayerWrapper.java
+++ b/extensions/media2/src/main/java/com/google/android/exoplayer2/ext/media2/PlayerWrapper.java
@@ -320,12 +320,12 @@ import java.util.List;
   }
 
   public boolean setPlaybackSpeed(float playbackSpeed) {
-    player.setPlaybackParameters(new PlaybackParameters(playbackSpeed));
+    player.setPlaybackSpeed(playbackSpeed);
     return true;
   }
 
   public float getPlaybackSpeed() {
-    return player.getPlaybackParameters().speed;
+    return player.getPlaybackSpeed();
   }
 
   public void reset() {
diff --git a/library/ui/src/main/java/com/google/android/exoplayer2/ui/PlayerNotificationManager.java b/library/ui/src/main/java/com/google/android/exoplayer2/ui/PlayerNotificationManager.java
index b85bbc64b1..d0e7b0da9e 100644
--- a/library/ui/src/main/java/com/google/android/exoplayer2/ui/PlayerNotificationManager.java
+++ b/library/ui/src/main/java/com/google/android/exoplayer2/ui/PlayerNotificationManager.java
@@ -923,7 +923,7 @@ public class PlayerNotificationManager {
    *   <li>The media is not {@link Player#isCurrentWindowDynamic() dynamically changing its
    *       duration} (like for example a live stream).
    *   <li>The media is not {@link Player#isPlayingAd() interrupted by an ad}.
-   *   <li>The media is played at {@link Player#getPlaybackParameters() regular speed}.
+   *   <li>The media is played at {@link Player#getPlaybackSpeed() regular speed}.
    *   <li>The device is running at least API 21 (Lollipop).
    * </ul>
    *
@@ -1086,7 +1086,7 @@ public class PlayerNotificationManager {
         && player.isPlaying()
         && !player.isPlayingAd()
         && !player.isCurrentWindowDynamic()
-        && player.getPlaybackParameters().speed == 1f) {
+        && player.getPlaybackSpeed() == 1f) {
       builder
           .setWhen(System.currentTimeMillis() - player.getContentPosition())
           .setShowWhen(true)