diff --git a/domain/src/main/java/com/github/nullptroma/wallenc/domain/interfaces/IStorageInfo.kt b/domain/src/main/java/com/github/nullptroma/wallenc/domain/interfaces/IStorageInfo.kt
index 400dea2..c958f3f 100644
--- a/domain/src/main/java/com/github/nullptroma/wallenc/domain/interfaces/IStorageInfo.kt
+++ b/domain/src/main/java/com/github/nullptroma/wallenc/domain/interfaces/IStorageInfo.kt
@@ -3,7 +3,7 @@ package com.github.nullptroma.wallenc.domain.interfaces
 import kotlinx.coroutines.flow.StateFlow
 import java.util.UUID
 
-interface IStorageInfo {
+sealed interface IStorageInfo {
     val size: StateFlow<Long?>
     val numberOfFiles: StateFlow<Int?>
     val uuid: UUID
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 54632bf..64e3c26 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("Cannot manage storage on StorageInfo")
-        _storage = storage
+        //if(storage !is IStorage)
+         //   throw Exception("Can not manage storage on StorageInfo")
+        _storage = storage as IStorage
     }
 
     suspend fun getAllFiles(): List<IFile> {