1 files changed, 5 insertions, 2 deletions
diff --git a/Controllers/MediaController.cs b/Controllers/MediaController.cs
index 8070199..fa6e7ab 100644
--- a/Controllers/MediaController.cs
+++ b/Controllers/MediaController.cs
@@ -1,6 +1,7 @@
 using HyperBooru.Services;
 using HyperBooru.Util;
 using Microsoft.AspNetCore.Mvc;
+using Microsoft.EntityFrameworkCore;
 
 namespace HyperBooru.Controllers;
 
@@ -23,13 +24,15 @@ public class MediaController : Controller {
 
     [HttpGet("{mediaId}")]
     public IActionResult Fetch([FromRoute] Guid mediaId) {
-        var media = db.Media.First(m => m.Guid == mediaId);
+        var media = db.Media
+            .Include(m => m.CurrentUploadedFile)
+            .First(m => m.Guid == mediaId);
         if(media is null)
             return NotFound();
 
         var fs = System.IO.File.OpenRead(mediaService.GetPath(media));
 
-        return new FileStreamResult(fs, media.MimeType);
+        return new FileStreamResult(fs, media.CurrentUploadedFile.MimeType);
     }
 
     [HttpGet("thumb/{mediaId}")]