1 files changed, 5 insertions, 0 deletions
diff --git a/main/Test.hs b/main/Test.hs
index f9657c2..6ee75bd 100644
--- a/main/Test.hs
+++ b/main/Test.hs
@@ -289,6 +289,11 @@ instance MonadHead LocalState CommandM where
         modify $ \s -> s { tsHead = Just h' }
         return x
 
+    getLocalHeadCache _ = do
+        h <- maybe (fail "failed to reload head") return =<< reloadHead =<< getOrLoadHead
+        modify $ \s -> s { tsHead = Just h }
+        return $ headCache h
+
 type Command = CommandM ()
 
 commands :: [ ( Text, Command ) ]