diff --git a/mozilla/netwerk/test/TestSocketTransport.cpp b/mozilla/netwerk/test/TestSocketTransport.cpp
index f49293fd66e..74f3a300285 100644
--- a/mozilla/netwerk/test/TestSocketTransport.cpp
+++ b/mozilla/netwerk/test/TestSocketTransport.cpp
@@ -307,8 +307,7 @@ TestConnection::TestConnection(const char* aHostName, PRInt32 aPort,
     if (NS_SUCCEEDED(rv)) {
       // Set up the notification callbacks to provide a progress event sink.
       // That way we exercise the progress notification proxy code.
-      nsCOMPtr<nsIProgressEventSink> sink = do_GetInterface((nsIInterfaceRequestor*)this);
-      rv = mTransport->SetProgressEventSink(sink);
+      mTransport->SetNotificationCallbacks((nsIInterfaceRequestor*)this, PR_FALSE);
     }
   }