1 files changed, 9 insertions, 1 deletions

diff --git a/src/symbol_table.c b/src/symbol_table.c
index 68e24f0..cff675a 100644
--- a/src/symbol_table.c
+++ b/src/symbol_table.c
@@ -217,17 +217,25 @@ static char *string_of_type(int type) {
     return "CHAR";
   case VOID_T:
     return "VOID";
+  case TAB:
+    return "TAB";
   default:
     return "UNEXPECTED";
   }
 }
 
-void check_expected_type(int type_to_check, int type_expected) {
+void check_expected_type(Type type_to_check, Type type_expected) {
   if (type_to_check != type_expected)
     fprintf(stderr, "Expected type : %s -> Got type : %s (near line %d)\n",
             string_of_type(type_expected), string_of_type(type_to_check),
             lineno);
 }
+void check_expected_types(Type type_to_check, Type type_expected1, Type type_expected2) {
+  if (type_to_check != type_expected1 && type_to_check != type_expected2)
+    fprintf(stderr, "Expected type : %s OR %s-> Got type : %s (near line %d)\n",
+            string_of_type(type_expected1), string_of_type(type_expected2), string_of_type(type_to_check),
+            lineno);
+}
 
 /* returns false if symbol can't be found too */
 bool is_read_only(const char name[], Scope scope) {