diff --git a/app/src/main/java/com/infomaniak/swisstransfer/ui/screen/newtransfer/importfiles/FilesDetailsScreen.kt b/app/src/main/java/com/infomaniak/swisstransfer/ui/screen/newtransfer/importfiles/FilesDetailsScreen.kt
index f46231c0c..e2cd4e7f8 100644
--- a/app/src/main/java/com/infomaniak/swisstransfer/ui/screen/newtransfer/importfiles/FilesDetailsScreen.kt
+++ b/app/src/main/java/com/infomaniak/swisstransfer/ui/screen/newtransfer/importfiles/FilesDetailsScreen.kt
@@ -64,6 +64,15 @@ fun FilesDetailsScreen(
     )
 }
 
+private fun getOnFileRemoveCallback(
+    withFileDelete: Boolean,
+    importFilesViewModel: ImportFilesViewModel,
+): ((String) -> Unit)? {
+    return if (withFileDelete) {
+        { importFilesViewModel.removeFileByUid(it) }
+    } else null
+}
+
 @Composable
 private fun FilesDetailsScreen(
     title: String = "",