From be363b11a298abb6fc6e972352c4f8044b1beeec Mon Sep 17 00:00:00 2001
From: Sam Mohr <sam@sammohr.dev>
Date: Sat, 26 Oct 2024 16:58:21 -0700
Subject: [PATCH] Fix test_gen incorrect macro invocations

---
 crates/compiler/test_gen/src/gen_return.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/crates/compiler/test_gen/src/gen_return.rs b/crates/compiler/test_gen/src/gen_return.rs
index ce9365e8da..dbf4b0387b 100644
--- a/crates/compiler/test_gen/src/gen_return.rs
+++ b/crates/compiler/test_gen/src/gen_return.rs
@@ -6,6 +6,12 @@ use crate::helpers::llvm::assert_evals_to;
 #[cfg(feature = "gen-dev")]
 use crate::helpers::dev::assert_evals_to;
 
+#[cfg(feature = "gen-llvm")]
+use crate::helpers::llvm::identity;
+
+#[cfg(feature = "gen-dev")]
+use crate::helpers::dev::identity;
+
 #[allow(unused_imports)]
 use indoc::indoc;
 #[allow(unused_imports)]
@@ -97,6 +103,7 @@ fn early_return_solo() {
         "#,
         RocStr::from("abc"),
         RocStr,
+        identity,
         true
     );
 
@@ -109,6 +116,7 @@ fn early_return_solo() {
         "#,
         123,
         i64,
+        identity,
         true
     );
 }