From 8a1fd13db4db3c5da84c50c720991da9bc5e6289 Mon Sep 17 00:00:00 2001
From: Masahiro Sakai <masahiro.sakai@gmail.com>
Date: Sun, 9 Aug 2026 18:13:43 +0900
Subject: [PATCH] fix QuantifierRef.sort to return the correct sort for lambda
 expressions

---
 cvc5_pythonic_api/cvc5_pythonic.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cvc5_pythonic_api/cvc5_pythonic.py b/cvc5_pythonic_api/cvc5_pythonic.py
index 44c0d3b..0ac5516 100644
--- a/cvc5_pythonic_api/cvc5_pythonic.py
+++ b/cvc5_pythonic_api/cvc5_pythonic.py
@@ -9038,7 +9038,9 @@ def as_ast(self):
         return self.ast
 
     def sort(self):
-        """Return the Boolean sort"""
+        """Return the Boolean sort or sort of Lambda."""
+        if self.is_lambda():
+            return _sort(self.ctx, self.as_ast())
         return BoolSort(self.ctx)
 
     def is_forall(self):