From 035c888d9be86d546c49569b370dbe1264844fa7 Mon Sep 17 00:00:00 2001
From: mre7a <68907585+mre7a@users.noreply.github.com>
Date: Thu, 16 Feb 2023 10:43:55 +0100
Subject: [PATCH] Update openpype/hosts/maya/api/workfile_template_builder.py

Co-authored-by: Jakub Trllo <43494761+iLLiCiTiT@users.noreply.github.com>
---
 openpype/hosts/maya/api/workfile_template_builder.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/openpype/hosts/maya/api/workfile_template_builder.py b/openpype/hosts/maya/api/workfile_template_builder.py
index 1e6094e996..094f45221c 100644
--- a/openpype/hosts/maya/api/workfile_template_builder.py
+++ b/openpype/hosts/maya/api/workfile_template_builder.py
@@ -50,6 +50,7 @@ class MayaTemplateBuilder(AbstractTemplateBuilder):
             return True
 
         # update imported sets information
+        asset_name = get_current_asset_name()
         for node in imported_sets:
             if not cmds.attributeQuery("id", node=node, exists=True):
                 continue
@@ -57,9 +58,9 @@ class MayaTemplateBuilder(AbstractTemplateBuilder):
                 continue
             if not cmds.attributeQuery("asset", node=node, exists=True):
                 continue
-            asset = legacy_io.Session["AVALON_ASSET"]
 
-            cmds.setAttr("{}.asset".format(node), asset, type="string")
+            cmds.setAttr(
+                "{}.asset".format(node), asset_name, type="string")
 
         return True