diff --git a/json_generator/main.py b/json_generator/main.py
index 3602d15..6cdee9d 100644
--- a/json_generator/main.py
+++ b/json_generator/main.py
@@ -25,7 +25,7 @@ def modify_variable(var, vars):
 		response = input("Select a property to modify for %s or q to exit property modifying: " %(var.name))
 		
 		if response == "type":
-			response = input("Enter the type of variable this is (currently %s): " %(expression.Variable.get_variable_type(var.type)))
+			response = input("Enter the type of variable this is (currently %s): " %(var.type))
 			vars[var.name].type = response
 			print("Updated to %s." %(response))
 			
diff --git a/main.py b/main.py
index 6f4493b..53cf367 100755
--- a/main.py
+++ b/main.py
@@ -31,7 +31,7 @@ def modify_variable(var, vars):
 		response = input("Select a property to modify for %s or q to exit property modifying: " %(var.name))
 		
 		if response == "type":
-			response = input("Enter the type of variable this is (currently %s): " %(expression.Variable.get_variable_type(var.type)))
+			response = input("Enter the type of variable this is (currently %s): " %(var.type))
 			vars[var.name].type = response
 			print("Updated to %s." %(response))
 			
@@ -74,7 +74,7 @@ rev_tokens = copy.deepcopy(tokens)
 
 for t in rev_tokens:
 	if t.type == token.TokenType.variable:
-		variables[t.value] = expression.Variable(t.value, expression.Variable.get_variable_type(t.value))
+		variables[t.value] = expression.Variable(t.value, expression.Variable.get_variable_type(t.value[0]))
 
 selected_variable = None