From 585fdbad396a089e11717eb6ce52e60a07e900e9 Mon Sep 17 00:00:00 2001
From: Patrick Barron <barronpm@gmail.com>
Date: Sun, 31 May 2020 17:00:57 -0400
Subject: [PATCH] Apply review suggestion and fix entity concurrency tokens.

---
 Jellyfin.Server.Implementations/JellyfinDb.cs | 1 +
 Jellyfin.Server/CoreAppHost.cs                | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/Jellyfin.Server.Implementations/JellyfinDb.cs b/Jellyfin.Server.Implementations/JellyfinDb.cs
index ae783efd84..d3c0267e84 100644
--- a/Jellyfin.Server.Implementations/JellyfinDb.cs
+++ b/Jellyfin.Server.Implementations/JellyfinDb.cs
@@ -81,6 +81,7 @@ namespace Jellyfin.Server.Implementations
         {
             foreach (var saveEntity in ChangeTracker.Entries()
                 .Where(e => e.State == EntityState.Modified)
+                .Select(entry => entry.Entity)
                 .OfType<ISavingChanges>())
             {
                 saveEntity.OnSavingChanges();
diff --git a/Jellyfin.Server/CoreAppHost.cs b/Jellyfin.Server/CoreAppHost.cs
index 716abcb637..fe07411a68 100644
--- a/Jellyfin.Server/CoreAppHost.cs
+++ b/Jellyfin.Server/CoreAppHost.cs
@@ -84,8 +84,11 @@ namespace Jellyfin.Server
         /// <inheritdoc />
         protected override IEnumerable<Assembly> GetAssembliesWithPartsInternal()
         {
+            // Jellyfin.Server
             yield return typeof(CoreAppHost).Assembly;
-            yield return Assembly.Load("Jellyfin.Server.Implementations");
+
+            // Jellyfin.Server.Implementations
+            yield return typeof(JellyfinDb).Assembly;
         }
 
         /// <inheritdoc />