diff --git a/main.py b/main.py
index fea2cf8..e1950bd 100644
--- a/main.py
+++ b/main.py
@@ -2,21 +2,19 @@ import json, sys, traceback
 
 def get_service_index():
     for index, value in enumerate(json_data['service']):
-        if(value['name'] == service_name):
+        if(value['name'] == sys.argv[2]):
             return index
     exit(traceback.print_exc())
     
 def set_latest_version():
     service_index = get_service_index()
-    json_data["service"][service_index]["latest_version"] = tagName
+    json_data["service"][service_index]["latest_version"] = sys.argv[1]
     
 if __name__ == '__main__':
     
     if len(sys.argv) != 3:
         exit(traceback.print_exc())
     
-    tagName = sys.argv[1]
-    service_name = sys.argv[2]
     file_path = "./candidate-version.json"
     
     with open(file_path, 'r') as json_file: