From e1caa6fff8a8ec0e0f6b4d012c0aaba779737e86 Mon Sep 17 00:00:00 2001
From: Xun Yang <x299yang@uwaterloo.ca>
Date: Mon, 13 Apr 2020 16:01:38 -0400
Subject: [PATCH] init short

---
 UnitNodes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/UnitNodes.py b/UnitNodes.py
index 9729e9e..34ee0c7 100644
--- a/UnitNodes.py
+++ b/UnitNodes.py
@@ -145,8 +145,8 @@ class LiteralNode(ASTNode):
             return
 
         # int
-        if self.name == 'int':
-            self.code += p("mov", "eax", str(self.getConstant()), " set to literal int")
+        if self.name in ['int', 'short']:
+            self.code += p("mov", "eax", str(self.getConstant()), " set to literal num")
             return
 
     def getConstant(self):
-- 
GitLab