diff --git a/pkg/twap/v2/stream_executor_test.go b/pkg/twap/v2/stream_executor_test.go
index fe3f05cdb..e53658efc 100644
--- a/pkg/twap/v2/stream_executor_test.go
+++ b/pkg/twap/v2/stream_executor_test.go
@@ -81,6 +81,9 @@ func bindMockMarketDataStream(mockStream *mocks.MockStream, stream *types.Standa
 	mockStream.EXPECT().OnConnect(Catch(func(x any) {
 		stream.OnConnect(x.(func()))
 	})).AnyTimes()
+	mockStream.EXPECT().OnDisconnect(Catch(func(x any) {
+		stream.OnDisconnect(x.(func()))
+	})).AnyTimes()
 }
 
 func bindMockUserDataStream(mockStream *mocks.MockStream, stream *types.StandardStream) {