diff --git a/app/src/test/java/org/mozilla/fenix/components/TestComponents.kt b/app/src/test/java/org/mozilla/fenix/components/TestComponents.kt
index fdc4fca9a..5a67c7edc 100644
--- a/app/src/test/java/org/mozilla/fenix/components/TestComponents.kt
+++ b/app/src/test/java/org/mozilla/fenix/components/TestComponents.kt
@@ -23,7 +23,8 @@ class TestComponents(private val context: Context) : Components(context) {
             core.sessionManager,
             core.store,
             search.searchEngineManager,
-            core.webAppShortcutManager
+            core.webAppShortcutManager,
+            core.topSiteStorage
         )
     }
     override val intentProcessors by lazy { mockk<IntentProcessors>(relaxed = true) }
diff --git a/app/src/test/java/org/mozilla/fenix/components/TestCore.kt b/app/src/test/java/org/mozilla/fenix/components/TestCore.kt
index a28fdd067..23c61d2db 100644
--- a/app/src/test/java/org/mozilla/fenix/components/TestCore.kt
+++ b/app/src/test/java/org/mozilla/fenix/components/TestCore.kt
@@ -14,6 +14,7 @@ import mozilla.components.concept.engine.Engine
 import mozilla.components.concept.engine.Settings
 import mozilla.components.concept.fetch.Client
 import mozilla.components.feature.pwa.WebAppShortcutManager
+import mozilla.components.feature.top.sites.DefaultTopSitesStorage
 import mozilla.components.support.base.crash.CrashReporting
 
 class TestCore(context: Context, crashReporter: CrashReporting) : Core(context, crashReporter) {
@@ -26,4 +27,5 @@ class TestCore(context: Context, crashReporter: CrashReporting) : Core(context,
     override val client = mockk<Client>()
     override val webAppShortcutManager = mockk<WebAppShortcutManager>()
     override val thumbnailStorage = mockk<ThumbnailStorage>()
+    override val topSiteStorage = mockk<DefaultTopSitesStorage>()
 }
diff --git a/app/src/test/java/org/mozilla/fenix/home/sessioncontrol/viewholders/TopSiteViewHolderTest.kt b/app/src/test/java/org/mozilla/fenix/home/sessioncontrol/viewholders/TopSiteViewHolderTest.kt
index 8477eb7d2..b2350f8c2 100644
--- a/app/src/test/java/org/mozilla/fenix/home/sessioncontrol/viewholders/TopSiteViewHolderTest.kt
+++ b/app/src/test/java/org/mozilla/fenix/home/sessioncontrol/viewholders/TopSiteViewHolderTest.kt
@@ -32,14 +32,17 @@ class TopSiteViewHolderTest {
 
     @Test
     fun `binds list of top sites`() {
-        TopSiteViewHolder(view, interactor).bind(listOf(
-            object : TopSite {
-                override val id = 1L
-                override val isDefault = true
-                override val title = "Pocket"
-                override val url = "https://getpocket.com"
-            }
-        ))
+        TopSiteViewHolder(view, interactor).bind(
+            listOf(
+                TopSite(
+                    id = 1L,
+                    title = "Pocket",
+                    url = "https://getpocket.com",
+                    createdAt = 0,
+                    type = TopSite.Type.DEFAULT
+                )
+            )
+        )
 
         assertEquals(1, view.top_sites_list.adapter!!.itemCount)
     }
diff --git a/app/src/test/java/org/mozilla/fenix/home/sessioncontrol/viewholders/topsites/TopSiteItemViewHolderTest.kt b/app/src/test/java/org/mozilla/fenix/home/sessioncontrol/viewholders/topsites/TopSiteItemViewHolderTest.kt
index 47c18277c..321bad7a9 100644
--- a/app/src/test/java/org/mozilla/fenix/home/sessioncontrol/viewholders/topsites/TopSiteItemViewHolderTest.kt
+++ b/app/src/test/java/org/mozilla/fenix/home/sessioncontrol/viewholders/topsites/TopSiteItemViewHolderTest.kt
@@ -22,12 +22,13 @@ class TopSiteItemViewHolderTest {
 
     private lateinit var view: View
     private lateinit var interactor: TopSiteInteractor
-    private val pocket = object : TopSite {
-        override val id = 1L
-        override val isDefault = true
-        override val title = "Pocket"
-        override val url = "https://getpocket.com"
-    }
+    private val pocket = TopSite(
+        id = 1L,
+        title = "Pocket",
+        url = "https://getpocket.com",
+        createdAt = 0,
+        type = TopSite.Type.DEFAULT
+    )
 
     @Before
     fun setup() {