diff --git a/generate_state_table.c b/generate_state_table.c
index bc9bc41..0648edc 100644
--- a/generate_state_table.c
+++ b/generate_state_table.c
@@ -47,7 +47,7 @@ INTERNAL i32 column_width = 0;
 INTERNAL lexer_state_t state_table[COUNT_LEXER_STATES][COUNT_LEXER_INPUTS] = {
     0};
 
-INTERNAL const char *input_strings[] = {
+INTERNAL const char *input_strings[COUNT_LEXER_STATES] = {
     [LEXER_INPUT_WHITE_SPACE] = "WHITE_SPACE",
     [LEXER_INPUT_OPEN_BRACE] = "OPEN_BRACE",
     [LEXER_INPUT_CLOSE_BRACE] = "CLOSE_BRACE",
@@ -84,7 +84,7 @@ INTERNAL const char *input_strings[] = {
     [LEXER_INPUT_OTHER] = "OTHER",
 };
 
-INTERNAL const char *state_strings[] = {
+INTERNAL const char *state_strings[COUNT_LEXER_STATES] = {
     [LEXER_STATE_ERROR] = "LEXER_STATE_ERROR",
     [LEXER_STATE_START] = "LEXER_STATE_START",
     [LEXER_STATE_VALUE] = "LEXER_STATE_VALUE",