diff --git a/DysonNetwork.Drive/Storage/CloudFile.cs b/DysonNetwork.Drive/Storage/CloudFile.cs
index 0225af4..7c8b358 100644
--- a/DysonNetwork.Drive/Storage/CloudFile.cs
+++ b/DysonNetwork.Drive/Storage/CloudFile.cs
@@ -102,6 +102,7 @@ public class CloudFile : ModelBase, ICloudFile, IIdentifiedResource
             FileMeta = GrpcTypeHelper.ConvertObjectToByteString(FileMeta),
             // Convert user metadata
             UserMeta = GrpcTypeHelper.ConvertObjectToByteString(UserMeta),
+            SensitiveMarks = GrpcTypeHelper.ConvertObjectToByteString(SensitiveMarks)
         };
 
         return proto;
diff --git a/DysonNetwork.Shared/Data/CloudFileReferenceObject.cs b/DysonNetwork.Shared/Data/CloudFileReferenceObject.cs
index 794b29d..2420442 100644
--- a/DysonNetwork.Shared/Data/CloudFileReferenceObject.cs
+++ b/DysonNetwork.Shared/Data/CloudFileReferenceObject.cs
@@ -1,6 +1,4 @@
-using System.Text.Json;
 using DysonNetwork.Shared.Proto;
-using Google.Protobuf;
 
 namespace DysonNetwork.Shared.Data;