Fix getLocalHeadCache in ReaderT MonadHead instance

Changelog: Fixed status indicator for new messages not being updated in some situations.

1 files changed, 4 insertions, 1 deletions

diff --git a/src/Erebos/State.hs b/src/Erebos/State.hs
index 3774d87..fefbffd 100644
--- a/src/Erebos/State.hs
+++ b/src/Erebos/State.hs
@@ -126,7 +126,10 @@ instance (HeadType a, MonadIO m) => MonadHead a (ReaderT (Head a) m) where
     updateLocalHead f = do
         h <- ask
         snd <$> updateHead' h (\h' -> local (const h') (f $ headStoredObject h'))
-    getLocalHeadCache _ = asks headCache
+
+    getLocalHeadCache _ = do
+        h <- ask
+        snd <$> updateHead' h (\h' -> return ( headStoredObject h', headCache h' ))
 
 
 newtype LocalHeadT h m a = LocalHeadT { runLocalHeadT_ :: Storage -> ( Stored h, HeadCacheType h ) -> m ( a, ( Stored h, HeadCacheType h ) ) }