diff --git a/app/src/androidTest/java/com/sadellie/unitto/screen/SwapUnitsTest.kt b/app/src/androidTest/java/com/sadellie/unitto/screen/SwapUnitsTest.kt
index 6eac2314..52c429e6 100644
--- a/app/src/androidTest/java/com/sadellie/unitto/screen/SwapUnitsTest.kt
+++ b/app/src/androidTest/java/com/sadellie/unitto/screen/SwapUnitsTest.kt
@@ -5,6 +5,7 @@ import android.content.Context
 import androidx.room.Room
 import androidx.test.core.app.ApplicationProvider
 import androidx.test.ext.junit.runners.AndroidJUnit4
+import com.sadellie.unitto.data.preferences.DataStoreModule
 import com.sadellie.unitto.data.preferences.UserPreferencesRepository
 import com.sadellie.unitto.data.units.ALL_UNITS
 import com.sadellie.unitto.data.units.MyUnitIDS
@@ -26,7 +27,7 @@ class SwapUnitsTest {
     fun setUp() {
         val context = ApplicationProvider.getApplicationContext<Context>()
         viewModel = MainViewModel(
-            UserPreferencesRepository(context),
+            UserPreferencesRepository(DataStoreModule().provideUserPreferencesDataStore(context)),
             MyBasedUnitsRepository(
                 Room.inMemoryDatabaseBuilder(
                     context,