diff --git a/app/src/main/java/org/futo/circles/di/DataSourceModule.kt b/app/src/main/java/org/futo/circles/di/DataSourceModule.kt
index c33c38e6683924e83dd13eef53d1992268699f7f..b44ca7940aab3a09488440fb1e46e55b90c979eb 100644
--- a/app/src/main/java/org/futo/circles/di/DataSourceModule.kt
+++ b/app/src/main/java/org/futo/circles/di/DataSourceModule.kt
@@ -31,6 +31,7 @@ import org.futo.circles.feature.sign_up.SignUpDataSource
 import org.futo.circles.feature.sign_up.setup_circles.SetupCirclesDataSource
 import org.futo.circles.feature.sign_up.setup_profile.SetupProfileDataSource
 import org.futo.circles.feature.sign_up.sign_up_type.SelectSignUpTypeDataSource
+import org.futo.circles.feature.sign_up.subscription_stage.SubscriptionStageDataSource
 import org.futo.circles.feature.sign_up.terms.AcceptTermsDataSource
 import org.futo.circles.feature.sign_up.validate_email.ValidateEmailDataSource
 import org.futo.circles.feature.sign_up.validate_token.ValidateTokenDataSource
@@ -92,4 +93,5 @@ val dataSourceModule = module {
     factory { UserOptionsDataSource() }
     factory { (userId: String) -> UserDataSource(get(), userId) }
     single { PickDeviceMediaDataSource(get()) }
+    factory { SubscriptionStageDataSource(get()) }
 }
\ No newline at end of file
diff --git a/app/src/main/java/org/futo/circles/di/UiModule.kt b/app/src/main/java/org/futo/circles/di/UiModule.kt
index fb43d9535d3fe0aebaa5aff01f779c5fd4b30963..e68fdd6476efeff7c1f4c8bb2c1af778d6bf3294 100644
--- a/app/src/main/java/org/futo/circles/di/UiModule.kt
+++ b/app/src/main/java/org/futo/circles/di/UiModule.kt
@@ -28,6 +28,7 @@ import org.futo.circles.feature.sign_up.SignUpViewModel
 import org.futo.circles.feature.sign_up.setup_circles.SetupCirclesViewModel
 import org.futo.circles.feature.sign_up.setup_profile.SetupProfileViewModel
 import org.futo.circles.feature.sign_up.sign_up_type.SelectSignUpTypeViewModel
+import org.futo.circles.feature.sign_up.subscription_stage.SubscriptionStageViewModel
 import org.futo.circles.feature.sign_up.terms.AcceptTermsViewModel
 import org.futo.circles.feature.sign_up.validate_email.ValidateEmailViewModel
 import org.futo.circles.feature.sign_up.validate_token.ValidateTokenViewModel
@@ -102,4 +103,5 @@ val uiModule = module {
     viewModel { (isVideoAvailable: Boolean) ->
         PickDeviceMediaViewModel(isVideoAvailable, get())
     }
+    viewModel { SubscriptionStageViewModel(get()) }
 }
\ No newline at end of file