From 51d78df83fc69df8e54cb72212a91576da8bf5b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Roman=20Smr=C5=BE?= <roman.smrz@seznam.cz>
Date: Thu, 5 Dec 2024 22:14:21 +0100
Subject: Arguments for user-defined functions

---
 src/Parser/Core.hs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src/Parser/Core.hs')

diff --git a/src/Parser/Core.hs b/src/Parser/Core.hs
index 57b2eb4..5fb4c5f 100644
--- a/src/Parser/Core.hs
+++ b/src/Parser/Core.hs
@@ -200,7 +200,8 @@ localState :: TestParser a -> TestParser a
 localState inner = do
     s <- get
     x <- inner
-    put s
+    s' <- get
+    put s { testNextTypeVar = testNextTypeVar s', testTypeUnif = testTypeUnif s' }
     return x
 
 toplevel :: (a -> Toplevel) -> TestParser a -> TestParser Toplevel
-- 
cgit v1.2.3