From 7251dcf2264c655cfa583c168beb18ec30bdc3a4 Mon Sep 17 00:00:00 2001
From: Robert <mail@rbeekman.nl>
Date: Sun, 25 Sep 2022 20:48:49 +0200
Subject: [PATCH] Migrations - Fix possible DB issue with migrator

---
 src/Artemis.Storage/Migrations/M0021GradientNodes.cs | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/Artemis.Storage/Migrations/M0021GradientNodes.cs b/src/Artemis.Storage/Migrations/M0021GradientNodes.cs
index 11f062bd7..4d701ef24 100644
--- a/src/Artemis.Storage/Migrations/M0021GradientNodes.cs
+++ b/src/Artemis.Storage/Migrations/M0021GradientNodes.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.Linq;
 using Artemis.Storage.Entities.Profile;
 using Artemis.Storage.Entities.Profile.Nodes;
@@ -63,13 +64,13 @@ public class M0021GradientNodes : IStorageMigration
     public void Apply(LiteRepository repository)
     {
         // Find all color gradient data bindings, there's no really good way to do this so infer it from the value
-        ILiteCollection<ProfileEntity> collection = repository.Database.GetCollection<ProfileEntity>();
-        foreach (ProfileEntity profileEntity in collection.FindAll())
+        List<ProfileEntity> profiles = repository.Query<ProfileEntity>().ToList();
+        foreach (ProfileEntity profileEntity in profiles)
         {
             foreach (LayerEntity layer in profileEntity.Layers)
                 MigrateDataBinding(layer.LayerBrush.PropertyGroup);
 
-            collection.Update(profileEntity);
+            repository.Update(profileEntity);
         }
     }