diff --git a/devops-service/db/mongo/models/stack_template/stack_template_ec2.rb b/devops-service/db/mongo/models/stack_template/stack_template_ec2.rb
index 3c159bf..371a6d4 100644
--- a/devops-service/db/mongo/models/stack_template/stack_template_ec2.rb
+++ b/devops-service/db/mongo/models/stack_template/stack_template_ec2.rb
@@ -26,9 +26,9 @@ module Devops
 
       class << self
         def create(attrs)
-          template = attrs['template_body']
-          attrs['template_url'] = generate_template_file_and_upload_to_storage(attrs['id'], template)
-          super(attrs)
+          model = super(attrs)
+          model.update_template_url
+          model
         end
       end