diff --git a/bank-api-library/library/src/test/java/net/gini/android/bank/api/BankApiDocumentRemoteSourceTest.kt b/bank-api-library/library/src/test/java/net/gini/android/bank/api/BankApiDocumentRemoteSourceTest.kt
index 72cea28a4..52f59efd5 100644
--- a/bank-api-library/library/src/test/java/net/gini/android/bank/api/BankApiDocumentRemoteSourceTest.kt
+++ b/bank-api-library/library/src/test/java/net/gini/android/bank/api/BankApiDocumentRemoteSourceTest.kt
@@ -13,6 +13,8 @@ import net.gini.android.bank.api.requests.ErrorEvent
 import net.gini.android.bank.api.requests.ResolvePaymentBody
 import net.gini.android.bank.api.response.ConfigurationResponse
 import net.gini.android.bank.api.response.ResolvePaymentResponse
+import net.gini.android.core.api.response.DocumentLayoutResponse
+import net.gini.android.core.api.response.DocumentPageResponse
 import net.gini.android.core.api.response.PaymentRequestResponse
 import net.gini.android.core.api.response.PaymentResponse
 import okhttp3.RequestBody
@@ -169,6 +171,20 @@ class BankApiDocumentRemoteSourceTest {
             return Response.success(null)
         }
 
+        override suspend fun getDocumentLayout(
+            bearer: Map<String, String>,
+            documentId: String
+        ): Response<DocumentLayoutResponse> {
+            return Response.success(null)
+        }
+
+        override suspend fun getDocumentPages(
+            bearer: Map<String, String>,
+            documentId: String
+        ): Response<List<DocumentPageResponse>> {
+            return Response.success(null)
+        }
+
         override suspend fun getPaymentRequests(bearer: Map<String, String>): Response<List<PaymentRequestResponse>> {
             // Is tested in core api library
             return Response.success(null)
diff --git a/bank-sdk/sdk/src/main/java/net/gini/android/bank/sdk/GiniBank.kt b/bank-sdk/sdk/src/main/java/net/gini/android/bank/sdk/GiniBank.kt
index 625acf493..2f8efbf66 100644
--- a/bank-sdk/sdk/src/main/java/net/gini/android/bank/sdk/GiniBank.kt
+++ b/bank-sdk/sdk/src/main/java/net/gini/android/bank/sdk/GiniBank.kt
@@ -308,6 +308,7 @@ object GiniBank {
         callback: (CreateCaptureFlowFragmentForIntentResult) -> Unit
     ): CancellationToken {
         check(giniCapture != null) { "Capture feature is not configured. Call setCaptureConfiguration before creating the CaptureFlowFragment." }
+        BankSdkIsolatedKoinContext.init(context)
         return giniCapture!!.createDocumentForImportedFiles(
             intent,
             context,
diff --git a/bank-sdk/sdk/src/main/java/net/gini/android/bank/sdk/di/BankSdkIsolatedKoinContext.kt b/bank-sdk/sdk/src/main/java/net/gini/android/bank/sdk/di/BankSdkIsolatedKoinContext.kt
index 681e22c2a..49d196429 100644
--- a/bank-sdk/sdk/src/main/java/net/gini/android/bank/sdk/di/BankSdkIsolatedKoinContext.kt
+++ b/bank-sdk/sdk/src/main/java/net/gini/android/bank/sdk/di/BankSdkIsolatedKoinContext.kt
@@ -20,7 +20,7 @@ object BankSdkIsolatedKoinContext {
     val koin: Koin by lazy {
         koinApplication {
             val ctx = context
-            check(ctx != null) {
+             check(ctx != null) {
                 "Koin needs to be initialized first. " +
                         "Call BankSdkIsolatedKoinContext.init(context)!"
             }
diff --git a/core-api-library/library/src/test/java/net/gini/android/core/api/DocumentRemoteSourceTest.kt b/core-api-library/library/src/test/java/net/gini/android/core/api/DocumentRemoteSourceTest.kt
index c0967f37e..52d88bf6d 100644
--- a/core-api-library/library/src/test/java/net/gini/android/core/api/DocumentRemoteSourceTest.kt
+++ b/core-api-library/library/src/test/java/net/gini/android/core/api/DocumentRemoteSourceTest.kt
@@ -8,6 +8,8 @@ import kotlinx.coroutines.test.StandardTestDispatcher
 import kotlinx.coroutines.test.TestScope
 import kotlinx.coroutines.test.advanceUntilIdle
 import kotlinx.coroutines.test.runTest
+import net.gini.android.core.api.response.DocumentLayoutResponse
+import net.gini.android.core.api.response.DocumentPageResponse
 import net.gini.android.core.api.response.PaymentRequestResponse
 import net.gini.android.core.api.test.DocumentRemoteSourceForTests
 import net.gini.android.core.api.test.MockGiniApiType
@@ -234,5 +236,21 @@ class DocumentRemoteSourceTest {
             bearerAuthHeader = bearer["Authorization"]
             return Response.success(null)
         }
+
+        override suspend fun getDocumentLayout(
+            bearer: Map<String, String>,
+            documentId: String
+        ): Response<DocumentLayoutResponse> {
+            bearerAuthHeader = bearer["Authorization"]
+            return Response.success(null)
+        }
+
+        override suspend fun getDocumentPages(
+            bearer: Map<String, String>,
+            documentId: String
+        ): Response<List<DocumentPageResponse>> {
+            bearerAuthHeader = bearer["Authorization"]
+            return Response.success(null)
+        }
     }
 }
\ No newline at end of file
diff --git a/health-api-library/library/src/test/java/net/gini/android/health/api/HealthApiDocumentRemoteSourceTest.kt b/health-api-library/library/src/test/java/net/gini/android/health/api/HealthApiDocumentRemoteSourceTest.kt
index 4f2a5252f..018881dc9 100644
--- a/health-api-library/library/src/test/java/net/gini/android/health/api/HealthApiDocumentRemoteSourceTest.kt
+++ b/health-api-library/library/src/test/java/net/gini/android/health/api/HealthApiDocumentRemoteSourceTest.kt
@@ -8,6 +8,8 @@ import kotlinx.coroutines.test.StandardTestDispatcher
 import kotlinx.coroutines.test.TestScope
 import kotlinx.coroutines.test.advanceUntilIdle
 import kotlinx.coroutines.test.runTest
+import net.gini.android.core.api.response.DocumentLayoutResponse
+import net.gini.android.core.api.response.DocumentPageResponse
 import net.gini.android.core.api.response.PaymentRequestResponse
 import net.gini.android.core.api.response.PaymentResponse
 import net.gini.android.health.api.models.PaymentRequestInput
@@ -225,5 +227,19 @@ class HealthApiDocumentRemoteSourceTest {
             return Response.success(null)
         }
 
+        override suspend fun getDocumentLayout(
+            bearer: Map<String, String>,
+            documentId: String
+        ): Response<DocumentLayoutResponse> {
+            return Response.success(null)
+        }
+
+        override suspend fun getDocumentPages(
+            bearer: Map<String, String>,
+            documentId: String
+        ): Response<List<DocumentPageResponse>> {
+            return Response.success(null)
+        }
+
     }
 }
\ No newline at end of file