diff --git a/testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeMediaSource.java b/testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeMediaSource.java
index f2478845aa..17613ce519 100644
--- a/testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeMediaSource.java
+++ b/testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeMediaSource.java
@@ -78,6 +78,7 @@ public class FakeMediaSource implements MediaSource {
       ExoPlayer player, boolean isTopLevelSource, Listener listener) {
     assertThat(preparedSource).isFalse();
     preparedSource = true;
+    releasedSource = false;
     this.listener = listener;
     sourceInfoRefreshHandler = new Handler();
     if (timeline != null) {
@@ -116,6 +117,10 @@ public class FakeMediaSource implements MediaSource {
     assertThat(releasedSource).isFalse();
     assertThat(activeMediaPeriods.isEmpty()).isTrue();
     releasedSource = true;
+    preparedSource = false;
+    sourceInfoRefreshHandler.removeCallbacksAndMessages(null);
+    sourceInfoRefreshHandler = null;
+    listener = null;
   }
 
   /**