diff --git a/domain/src/main/java/com/github/nullptroma/wallenc/domain/usecases/StorageFileManagementUseCase.kt b/domain/src/main/java/com/github/nullptroma/wallenc/domain/usecases/StorageFileManagementUseCase.kt
index 64e3c26..55d7e94 100644
--- a/domain/src/main/java/com/github/nullptroma/wallenc/domain/usecases/StorageFileManagementUseCase.kt
+++ b/domain/src/main/java/com/github/nullptroma/wallenc/domain/usecases/StorageFileManagementUseCase.kt
@@ -9,9 +9,9 @@ class StorageFileManagementUseCase {
     private var _storage: IStorage? = null
 
     fun setStorage(storage: IStorageInfo) {
-        //if(storage !is IStorage)
-         //   throw Exception("Can not manage storage on StorageInfo")
-        _storage = storage as IStorage
+        when (storage) {
+            is IStorage -> _storage = storage
+        }
     }
 
     suspend fun getAllFiles(): List<IFile> {