From fb52c53552b1bb4ae92fcfb715b5627a334c5af9 Mon Sep 17 00:00:00 2001
From: Aiden McClelland <me@drbonez.dev>
Date: Wed, 26 May 2021 15:07:11 -0600
Subject: [PATCH] fix option get_mut

---
 patch-db/src/model.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/patch-db/src/model.rs b/patch-db/src/model.rs
index b0a97a9..fdc1137 100644
--- a/patch-db/src/model.rs
+++ b/patch-db/src/model.rs
@@ -210,7 +210,10 @@ impl<T: HasModel + Serialize + for<'de> Deserialize<'de>> OptionModel<T> {
         Ok(ModelData(db.get(self.0.as_ref()).await?))
     }
 
-    pub async fn get_mut<Db: DbHandle>(&self, db: &mut Db) -> Result<ModelDataMut<T>, Error> {
+    pub async fn get_mut<Db: DbHandle>(
+        &self,
+        db: &mut Db,
+    ) -> Result<ModelDataMut<Option<T>>, Error> {
         self.lock(db, LockType::Write).await;
         let original = db.get_value(self.0.as_ref(), None).await?;
         let current = serde_json::from_value(original.clone())?;