diff --git a/smart_service/apis/publish_api.py b/smart_service/apis/publish_api.py
index f42e3e0..deb01d9 100644
--- a/smart_service/apis/publish_api.py
+++ b/smart_service/apis/publish_api.py
@@ -499,9 +499,10 @@ def new_publish(args, publish_type, vehicle, language, version):
 
                     file_flag, final_update_file = merge_json_files(
                         prev_full_update_file, file_name, full_update_file_name)
-                    get_step_total_count(final_update_file)
-                    set_publish_flag(publish_type, vehicle, language)
+                    
                     if file_flag:
+                        get_step_total_count(final_update_file)
+                        set_publish_flag(publish_type, vehicle, language)
                         return True, file_name.split("public")[1]
                         # return True, file_name
                     else:
@@ -527,9 +528,10 @@ def new_publish(args, publish_type, vehicle, language, version):
                     file_flag, final_update_file = merge_json_files(
                         prev_full_internal_update_file, file_name, full_update_file_name)
 
-                    get_step_total_count(final_update_file)
-                    set_publish_flag(publish_type, vehicle, language)
+                    
                     if file_flag:
+                        get_step_total_count(final_update_file)
+                        set_publish_flag(publish_type, vehicle, language)
                         return True, file_name.split("public")[1]
                         # return True, file_name
                     else:
@@ -572,9 +574,10 @@ def new_publish(args, publish_type, vehicle, language, version):
 
                     file_flag, final_update_file = merge_json_files(
                         prev_full_update_file, file_name, full_update_file_name)
-                    get_step_total_count(final_update_file)
-                    set_publish_flag(publish_type, vehicle, language)
+                    
                     if file_flag:
+                        get_step_total_count(final_update_file)
+                        set_publish_flag(publish_type, vehicle, language)
                         return True, file_name.split("public")[1]
                         # return True, file_name
                     else: