diff --git a/crates/aiken-project/src/snapshots/aiken_project__export__tests__basic_export.snap b/crates/aiken-project/src/snapshots/aiken_project__export__tests__basic_export.snap
index 03ce0b29..ed1fd9f8 100644
--- a/crates/aiken-project/src/snapshots/aiken_project__export__tests__basic_export.snap
+++ b/crates/aiken-project/src/snapshots/aiken_project__export__tests__basic_export.snap
@@ -1,6 +1,7 @@
 ---
 source: crates/aiken-project/src/export.rs
 description: "Code:\n\npub fn add(a: Int, b: Int) -> Int {\n    a + b\n}\n"
+snapshot_kind: text
 ---
 {
   "name": "test_module.add",
@@ -18,6 +19,12 @@ description: "Code:\n\npub fn add(a: Int, b: Int) -> Int {\n    a + b\n}\n"
       }
     }
   ],
+  "return_type": {
+    "title": "return_type",
+    "schema": {
+      "$ref": "#/definitions/Int"
+    }
+  },
   "compiledCode": "500101002322337000046eb4004dd68009",
   "hash": "b8374597a772cef80d891b7f6a03588e10cc19b780251228ba4ce9c6",
   "definitions": {
diff --git a/crates/aiken-project/src/snapshots/aiken_project__export__tests__recursive_types.snap b/crates/aiken-project/src/snapshots/aiken_project__export__tests__recursive_types.snap
index e15b7e66..586b057c 100644
--- a/crates/aiken-project/src/snapshots/aiken_project__export__tests__recursive_types.snap
+++ b/crates/aiken-project/src/snapshots/aiken_project__export__tests__recursive_types.snap
@@ -1,6 +1,7 @@
 ---
 source: crates/aiken-project/src/export.rs
 description: "Code:\n\npub type Foo<a> {\n  Empty\n  Bar(a, Foo<a>)\n}\n\npub fn add(a: Foo<Int>, b: Foo<Int>) -> Int {\n  when (a, b) is {\n    (Empty, Empty) -> 0\n    (Bar(x, y), Bar(c, d)) -> x + c + add(y, d)\n    (Empty, Bar(c, d)) -> c + add(Empty, d)\n    (Bar(x, y), Empty) -> x + add(y, Empty)\n  }\n}\n"
+snapshot_kind: text
 ---
 {
   "name": "test_module.add",
@@ -18,6 +19,12 @@ description: "Code:\n\npub type Foo<a> {\n  Empty\n  Bar(a, Foo<a>)\n}\n\npub fn
       }
     }
   ],
+  "return_type": {
+    "title": "return_type",
+    "schema": {
+      "$ref": "#/definitions/Int"
+    }
+  },
   "compiledCode": "59017d0101003232323232322232323232325333008300430093754002264a666012600a60146ea800452000132337006eb4c038004cc011300103d8798000300e300f001300b37540026018601a00a264a66601266e1d2002300a37540022646466e00cdc01bad300f002375a601e0026600a601e6020004601e602000260186ea8008c02cdd500109919b80375a601c00266008601c601e002980103d8798000300b37540046018601a00a601600860020024446464a666014600c60166ea80044c94ccc02cc01cc030dd50008a400026466e00dd69808000999803803a60103d879800030103011001300d3754002601c601e004264a66601666e1d2002300c37540022646466e00cdc01bad3011002375a60220026660100106022602400460226024002601c6ea8008c034dd500109919b80375a602000266600e00e60206022002980103d8798000300d3754004601c601e004601a002660160046601600297ae0370e90001980300119803000a5eb815cd2ab9d5573cae815d0aba201",
   "hash": "c6af3f04e300cb8c1d0429cc0d8e56a0413eef9fcb338f72076b426c",
   "definitions": {