diff --git a/DysonNetwork.Sphere/Chat/ChatController.cs b/DysonNetwork.Sphere/Chat/ChatController.cs
index caa145a..b1c4c92 100644
--- a/DysonNetwork.Sphere/Chat/ChatController.cs
+++ b/DysonNetwork.Sphere/Chat/ChatController.cs
@@ -125,7 +125,6 @@ public partial class ChatController(AppDatabase db, ChatService cs, ChatRoomServ
             .Include(m => m.Sender)
             .Include(m => m.Sender.Account)
             .Include(m => m.Sender.Account.Profile)
-            .Include(m => m.Attachments)
             .FirstOrDefaultAsync();
 
         if (message is null) return NotFound();
diff --git a/DysonNetwork.Sphere/Storage/FileService.ReferenceMigration.cs b/DysonNetwork.Sphere/Storage/FileService.ReferenceMigration.cs
index 5476ad9..8809036 100644
--- a/DysonNetwork.Sphere/Storage/FileService.ReferenceMigration.cs
+++ b/DysonNetwork.Sphere/Storage/FileService.ReferenceMigration.cs
@@ -1,3 +1,4 @@
+using System.ComponentModel.DataAnnotations;
 using Microsoft.EntityFrameworkCore;
 
 namespace DysonNetwork.Sphere.Storage;