diff --git a/dom/media/gtest/TestAudioTrackGraph.cpp b/dom/media/gtest/TestAudioTrackGraph.cpp
index 8b9ad9359aa5..2d3e52fb7536 100644
--- a/dom/media/gtest/TestAudioTrackGraph.cpp
+++ b/dom/media/gtest/TestAudioTrackGraph.cpp
@@ -529,7 +529,7 @@ TEST(TestAudioTrackGraph, AudioInputTrackDisabling)
     return graph->NotifyWhenDeviceStarted(inputTrack);
   });
 
-  MockCubebStream* stream = WaitFor(cubeb->StreamInitEvent());
+  RefPtr<SmartMockCubebStream> stream = WaitFor(cubeb->StreamInitEvent());
   EXPECT_TRUE(stream->mHasInput);
   Unused << WaitFor(p);