diff --git a/src/main/java/ru/soune/nocopy/service/file/impl/FileUploadServiceImpl.java b/src/main/java/ru/soune/nocopy/service/file/impl/FileUploadServiceImpl.java
index 8511493..db52626 100644
--- a/src/main/java/ru/soune/nocopy/service/file/impl/FileUploadServiceImpl.java
+++ b/src/main/java/ru/soune/nocopy/service/file/impl/FileUploadServiceImpl.java
@@ -249,7 +249,9 @@ public class FileUploadServiceImpl implements FileUploadService {
             cleanupSessionFiles(session);
 
             if (status != FileStatus.TEMP) {
-                noCopyFileService.addFile(fileUtil.createFileInfo(fileEntity));
+                FileProtector.FileInfo fileInfo = fileUtil.createFileInfo(fileEntity);
+
+                noCopyFileService.addFile(fileInfo);
             }
 
             log.info("File processing completed for session: {}", session.getUploadId());
diff --git a/src/main/java/ru/soune/nocopy/util/FileUtil.java b/src/main/java/ru/soune/nocopy/util/FileUtil.java
index 192e1d3..7bfb07b 100644
--- a/src/main/java/ru/soune/nocopy/util/FileUtil.java
+++ b/src/main/java/ru/soune/nocopy/util/FileUtil.java
@@ -41,6 +41,8 @@ public class FileUtil {
             return FileProtector.Type.VIDEO;
         } else if (mimeType.startsWith("audio")) {
             return FileProtector.Type.AUDIO;
+        } else if (mimeType.startsWith("document")) {
+            return FileProtector.Type.DOC;
         } else {
             return FileProtector.Type.IMAGE;
         }