1 files changed, 17 insertions, 22 deletions
diff --git a/java/dagger/hilt/android/internal/managers/ActivityRetainedComponentManager.java b/java/dagger/hilt/android/internal/managers/ActivityRetainedComponentManager.java
index 0af3dcd11..2d4a72bdd 100644
--- a/java/dagger/hilt/android/internal/managers/ActivityRetainedComponentManager.java
+++ b/java/dagger/hilt/android/internal/managers/ActivityRetainedComponentManager.java
@@ -18,8 +18,6 @@ package dagger.hilt.android.internal.managers;
 
 import androidx.lifecycle.ViewModel;
 import androidx.lifecycle.ViewModelProvider;
-import androidx.lifecycle.ViewModelStoreOwner;
-import android.content.Context;
 import androidx.annotation.NonNull;
 import androidx.annotation.Nullable;
 import androidx.activity.ComponentActivity;
@@ -84,26 +82,23 @@ final class ActivityRetainedComponentManager
   private final Object componentLock = new Object();
 
   ActivityRetainedComponentManager(ComponentActivity activity) {
-    this.viewModelProvider = getViewModelProvider(activity, activity.getApplication());
-  }
-
-  private ViewModelProvider getViewModelProvider(
-      ViewModelStoreOwner owner, Context applicationContext) {
-    return new ViewModelProvider(
-        owner,
-        new ViewModelProvider.Factory() {
-          @NonNull
-          @Override
-          @SuppressWarnings("unchecked")
-          public <T extends ViewModel> T create(@NonNull Class<T> aClass) {
-            ActivityRetainedComponent component =
-                EntryPoints.get(
-                        applicationContext, ActivityRetainedComponentBuilderEntryPoint.class)
-                    .retainedComponentBuilder()
-                    .build();
-            return (T) new ActivityRetainedComponentViewModel(component);
-          }
-        });
+    this.viewModelProvider =
+        new ViewModelProvider(
+            activity,
+            new ViewModelProvider.Factory() {
+              @NonNull
+              @Override
+              @SuppressWarnings("unchecked")
+              public <T extends ViewModel> T create(@NonNull Class<T> aClass) {
+                ActivityRetainedComponent component =
+                    EntryPoints.get(
+                            activity.getApplication(),
+                            ActivityRetainedComponentBuilderEntryPoint.class)
+                        .retainedComponentBuilder()
+                        .build();
+                return (T) new ActivityRetainedComponentViewModel(component);
+              }
+            });
   }
 
   @Override