diff --git a/chromium_src/components/optimization_guide/core/delivery/prediction_manager.cc b/chromium_src/components/optimization_guide/core/delivery/prediction_manager.cc
index bd8cce37fb8..848a207ff83 100644
--- a/chromium_src/components/optimization_guide/core/delivery/prediction_manager.cc
+++ b/chromium_src/components/optimization_guide/core/delivery/prediction_manager.cc
@@ -97,8 +97,8 @@ PredictionManager::GetOnDeviceSupplementaryModelsInfoForWebUI() const {
 }
 
 void PredictionManager::MaybeInitializeModelDownloads(
-    PrefService* local_state,
-    download::BackgroundDownloadService* background_download_service) {}
+    ProfileDownloadServiceTracker& profile_download_service_tracker,
+    PrefService* local_state) {}
 
 base::FilePath PredictionManager::GetBaseModelDirForDownload(
     proto::OptimizationTarget optimization_target) {