From 46a2aa2ffc71820767f0cdaead84c26dc240c893 Mon Sep 17 00:00:00 2001
From: Bruce Hill <bruce@bruce-hill.com>
Date: Thu, 12 Sep 2024 14:27:13 -0400
Subject: Fix up comprehensions so set comprehensions work and everything is a
 bit more clean

---
 typecheck.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'typecheck.c')

diff --git a/typecheck.c b/typecheck.c
index 192a335b..4952fbd7 100644
--- a/typecheck.c
+++ b/typecheck.c
@@ -1250,7 +1250,10 @@ type_t *get_type(env_t *env, ast_t *ast)
 
     case While: case For: return Type(VoidType);
     case InlineCCode: {
-        type_ast_t *type_ast = Match(ast, InlineCCode)->type;
+        auto inline_code = Match(ast, InlineCCode);
+        if (inline_code->type)
+            return inline_code->type;
+        type_ast_t *type_ast = inline_code->type_ast;
         return type_ast ? parse_type_ast(env, type_ast) : Type(VoidType);
     }
     case Unknown: code_err(ast, "I can't figure out the type of: %W", ast);
-- 
cgit v1.2.3