diff --git a/app/src/main/java/com/infomaniak/swisstransfer/di/SwissTransferInjectionModule.kt b/app/src/main/java/com/infomaniak/swisstransfer/di/SwissTransferInjectionModule.kt
index 5e7e969e0..b2584e158 100644
--- a/app/src/main/java/com/infomaniak/swisstransfer/di/SwissTransferInjectionModule.kt
+++ b/app/src/main/java/com/infomaniak/swisstransfer/di/SwissTransferInjectionModule.kt
@@ -18,16 +18,15 @@
 package com.infomaniak.swisstransfer.di
 
 import com.infomaniak.core.appintegrity.AppIntegrityManager
+import com.infomaniak.core.network.ApiEnvironment
 import com.infomaniak.multiplatform_swisstransfer.SharedApiUrlCreator
 import com.infomaniak.multiplatform_swisstransfer.SwissTransferInjection
-import com.infomaniak.multiplatform_swisstransfer.common.utils.ApiEnvironment
 import com.infomaniak.multiplatform_swisstransfer.managers.AccountManager
 import com.infomaniak.multiplatform_swisstransfer.managers.AppSettingsManager
 import com.infomaniak.multiplatform_swisstransfer.managers.FileManager
 import com.infomaniak.multiplatform_swisstransfer.managers.InMemoryUploadManager
 import com.infomaniak.multiplatform_swisstransfer.managers.TransferManager
 import com.infomaniak.multiplatform_swisstransfer.managers.UploadV2Manager
-import com.infomaniak.swisstransfer.BuildConfig
 import com.infomaniak.swisstransfer.upload.UploadSessionStarter
 import com.infomaniak.swisstransfer.upload.UploadSessionStarterV1
 import com.infomaniak.swisstransfer.upload.UploadSessionStarterV2
@@ -36,6 +35,7 @@ import dagger.Provides
 import dagger.hilt.InstallIn
 import dagger.hilt.components.SingletonComponent
 import javax.inject.Singleton
+import com.infomaniak.multiplatform_swisstransfer.common.utils.ApiEnvironment as KmpApiEnvironement
 
 @Module
 @InstallIn(SingletonComponent::class)
@@ -45,7 +45,7 @@ object SwissTransferInjectionModule {
     @Singleton
     fun provideSwissTransferInjection(@UserAgent userAgent: String): SwissTransferInjection {
         return SwissTransferInjection(
-            environment = if (BuildConfig.FLAVOR == "preprod") ApiEnvironment.Preprod else ApiEnvironment.Prod,
+            environment = ApiEnvironment.current.toKmpApiEnvironment(),
             userAgent = userAgent,
             crashReport = crashReport,
             databaseNameOrPath = "swisstransfer",
@@ -96,3 +96,11 @@ object SwissTransferInjectionModule {
     @Singleton
     fun provideSharedApiUrlCreator(sti: SwissTransferInjection): SharedApiUrlCreator = sti.sharedApiUrlCreator
 }
+
+private fun ApiEnvironment.toKmpApiEnvironment(): KmpApiEnvironement {
+    return when (this) {
+        ApiEnvironment.PreProd -> KmpApiEnvironement.Preprod
+        ApiEnvironment.Prod -> KmpApiEnvironement.Prod
+        is ApiEnvironment.Custom -> error("Custom api environment is not handled by the kmp lib")
+    }
+}