From 9addfdb80e4bbc52a8fc4c59d41865dac37300ad Mon Sep 17 00:00:00 2001
From: cyndichin <cgc95@cornell.edu>
Date: Fri, 15 May 2026 14:45:43 -0400
Subject: [PATCH] Bugfix FXIOS-15822 [Unit Test] fix top sites

---
 firefox-ios/Providers/TopSitesProvider.swift | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/firefox-ios/Providers/TopSitesProvider.swift b/firefox-ios/Providers/TopSitesProvider.swift
index f3f8954d45724..092d43dba0d76 100644
--- a/firefox-ios/Providers/TopSitesProvider.swift
+++ b/firefox-ios/Providers/TopSitesProvider.swift
@@ -2,6 +2,7 @@
 // License, v. 2.0. If a copy of the MPL was not distributed with this
 // file, You can obtain one at http://mozilla.org/MPL/2.0/
 
+import Common
 import Foundation
 import Shared
 import UIKit
@@ -36,9 +37,10 @@ extension TopSitesProvider {
     }
 }
 
-final class TopSitesProviderImplementation: TopSitesProvider, FeatureFlaggable {
+final class TopSitesProviderImplementation: TopSitesProvider {
     private let pinnedSiteFetcher: PinnedSites
     private let placesFetcher: RustPlaces
+    let featureFlagsProvider: FeatureFlagProviding
     private let prefs: Prefs
 
     @MainActor
@@ -60,11 +62,13 @@ final class TopSitesProviderImplementation: TopSitesProvider, FeatureFlaggable {
     init(
         placesFetcher: RustPlaces,
         pinnedSiteFetcher: PinnedSites,
-        prefs: Prefs
+        prefs: Prefs,
+        featureFlagsProvider: FeatureFlagProviding = AppContainer.shared.resolve()
     ) {
         self.placesFetcher = placesFetcher
         self.pinnedSiteFetcher = pinnedSiteFetcher
         self.prefs = prefs
+        self.featureFlagsProvider = featureFlagsProvider
     }
 
     func getTopSites(numberOfMaxItems: Int,