From 173154e4064ddf87915efc2614c72a4b996aa047 Mon Sep 17 00:00:00 2001
From: microproofs <kwhitemsg@gmail.com>
Date: Wed, 17 May 2023 17:56:01 -0400
Subject: [PATCH] fix: (code gen): expect type now properly tracks constructor
 dependency. This was specific to code gen generated functions.

---
 CHANGELOG.md                      | 2 ++
 crates/aiken-lang/src/gen_uplc.rs | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 927e2e5e..ccc869bb 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -6,6 +6,8 @@
 
 - **aiken-lang**: Fix for cases where identity function is used as a param to a
   function or assigned to a var.
+- **aiken-lang**: Fix for free unique caused by code gen function having a
+  missing dependency.
 
 ## v1.0.5-alpha - 2023-05-14
 
diff --git a/crates/aiken-lang/src/gen_uplc.rs b/crates/aiken-lang/src/gen_uplc.rs
index 8dcadf96..b90de927 100644
--- a/crates/aiken-lang/src/gen_uplc.rs
+++ b/crates/aiken-lang/src/gen_uplc.rs
@@ -2600,6 +2600,8 @@ impl<'a> CodeGenerator<'a> {
                 );
             } else if let Some(counter) = defined_data_types.get_mut(&data_type_name) {
                 *counter += 1;
+            } else {
+                defined_data_types.insert(data_type_name.clone(), 1);
             }
 
             func_stack.var(
@@ -2792,6 +2794,7 @@ impl<'a> CodeGenerator<'a> {
 
                 *dep_scope = dep_scope.common_ancestor(&func_scope);
             }
+
             dependency_map.insert(function.0, function.1);
         }