From 3d84003941ab4cab92d9750dc1ed815d9ba9957b Mon Sep 17 00:00:00 2001
From: Matt Brown <github@muglug.com>
Date: Mon, 8 May 2023 21:11:40 -0400
Subject: [PATCH] Fix incorrect codegen type

---
 src/BuildSchemaCLI.php                 | 2 +-
 tests/fixtures/expected_schema.codegen | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/BuildSchemaCLI.php b/src/BuildSchemaCLI.php
index 3a14443..cdf9386 100644
--- a/src/BuildSchemaCLI.php
+++ b/src/BuildSchemaCLI.php
@@ -87,7 +87,7 @@ public static function getRenderedHackTableSchemaWithClusters(
 		$file_contents .= "use type Slack\\SQLFake\\{Column, DataType, Index, TableSchema};\n";
 		$file_contents .= "\n";
 		$file_contents .= "<<__Memoize>>\n";
-		$file_contents .= "function {$function_name}(): dict<string, dict<string, table_schema>> {\n";
+		$file_contents .= "function {$function_name}(): dict<string, dict<string, TableSchema>> {\n";
 		$file_contents .= "\treturn dict[\n";
 		foreach ($table_schemas as $cluster => $tables) {
 			$file_contents .= "\t\t'{$cluster}' => ".self::getRenderedHackTableSchema($tables, "\t\t");
diff --git a/tests/fixtures/expected_schema.codegen b/tests/fixtures/expected_schema.codegen
index 9a52238..0fb8817 100644
--- a/tests/fixtures/expected_schema.codegen
+++ b/tests/fixtures/expected_schema.codegen
@@ -1,7 +1,7 @@
 use type Slack\SQLFake\{Column, DataType, Index, TableSchema};
 
 <<__Memoize>>
-function get_my_table_schemas(): dict<string, dict<string, table_schema>> {
+function get_my_table_schemas(): dict<string, dict<string, TableSchema>> {
 	return dict[
 		'prod' => dict[
 			'test' => new TableSchema(