1 files changed, 12 insertions, 0 deletions
diff --git a/main.c b/main.c
index 1657259..9f4565e 100644
--- a/main.c
+++ b/main.c
@@ -173,6 +173,18 @@ int size, *nxt;
 
 		*nxt = t->attr.nval->var_type;
 		return size;
+	} else if (t->type == INUM) {
+		if (--size == -1)
+			yyerror("VARIABLE COUNT CHANGED!!!\n");
+
+		*nxt = INT;
+		return size;
+	} else if (t->type == RNUM) {
+		if (--size == -1)
+			yyerror("VARIABLE COUNT CHANGED!!!\n");
+
+		*nxt = REAL;
+		return size;
 	}
 	return size;
 }