From 8c50de9b62b08bd59c34ec425b0b13ba7158c67f Mon Sep 17 00:00:00 2001
From: DenBond7 <DenBond7@gmail.com>
Date: Mon, 7 Mar 2022 18:01:10 +0200
Subject: [PATCH] Refactored code.| #1704

---
 .../jetpack/viewmodel/DownloadAttachmentViewModel.kt     | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/FlowCrypt/src/main/java/com/flowcrypt/email/jetpack/viewmodel/DownloadAttachmentViewModel.kt b/FlowCrypt/src/main/java/com/flowcrypt/email/jetpack/viewmodel/DownloadAttachmentViewModel.kt
index d8824185c2..6f0b355fb0 100644
--- a/FlowCrypt/src/main/java/com/flowcrypt/email/jetpack/viewmodel/DownloadAttachmentViewModel.kt
+++ b/FlowCrypt/src/main/java/com/flowcrypt/email/jetpack/viewmodel/DownloadAttachmentViewModel.kt
@@ -74,11 +74,10 @@ class DownloadAttachmentViewModel(val attachmentInfo: AttachmentInfo, applicatio
           }
         }
 
-        val email =
-          attachmentInfo.email ?: return@withContext Result.success(byteArrayOf())
-        val account = getAccountEntityWithDecryptedInfoSuspend(
-          roomDatabase.accountDao().getAccount(email)
-        ) ?: return@withContext Result.success(byteArrayOf())
+        val email = requireNotNull(attachmentInfo.email)
+        val account = requireNotNull(
+          getAccountEntityWithDecryptedInfoSuspend(roomDatabase.accountDao().getAccount(email))
+        )
 
         if (account.useAPI) {
           when (account.accountType) {