diff --git a/app/src/androidTest/java/com/google/android/samples/socialite/repository/TestRepository.kt b/app/src/androidTest/java/com/google/android/samples/socialite/repository/TestRepository.kt
index 4761808e..4c5a01ee 100644
--- a/app/src/androidTest/java/com/google/android/samples/socialite/repository/TestRepository.kt
+++ b/app/src/androidTest/java/com/google/android/samples/socialite/repository/TestRepository.kt
@@ -33,5 +33,6 @@ fun createTestRepository(): ChatRepository {
         notificationHelper = NotificationHelper(context),
         widgetModelRepository = WidgetModelRepository(database.widgetDao(), CoroutineScope(SupervisorJob() + Dispatchers.Default), context),
         coroutineScope = CoroutineScope(SupervisorJob() + Dispatchers.Default),
+        appContext = InstrumentationRegistry.getInstrumentation().targetContext,
     )
 }