diff --git a/xCAT-test/autotest/testcase/installation/reg_linux_statelite_installation_flat b/xCAT-test/autotest/testcase/installation/reg_linux_statelite_installation_flat
index dcb2303b3..8aa74ba8a 100644
--- a/xCAT-test/autotest/testcase/installation/reg_linux_statelite_installation_flat
+++ b/xCAT-test/autotest/testcase/installation/reg_linux_statelite_installation_flat
@@ -57,7 +57,7 @@ check:rc==0
 cmd:liteimg  __GETNODEATTR($$CN,os)__-__GETNODEATTR($$CN,arch)__-statelite-compute
 check:rc==0
 
-cmd:nodeset $$CN statelite
+cmd:nodeset $$CN osimage=__GETNODEATTR($$CN,os)__-__GETNODEATTR($$CN,arch)__-statelite-compute
 check:rc==0
 cmd:lsdef $$CN |grep provmethod
 check:rc==0
diff --git a/xCAT-test/autotest/testcase/installation/reg_linux_statelite_installation_hierarchy b/xCAT-test/autotest/testcase/installation/reg_linux_statelite_installation_hierarchy
index 06a5f66f1..ddbd24ebc 100644
--- a/xCAT-test/autotest/testcase/installation/reg_linux_statelite_installation_hierarchy
+++ b/xCAT-test/autotest/testcase/installation/reg_linux_statelite_installation_hierarchy
@@ -69,7 +69,7 @@ check:rc==0
 cmd:prsync /install $$SN:/
 check:rc==0
 
-cmd:nodeset $$CN statelite
+cmd:nodeset $$CN osimage=__GETNODEATTR($$CN,os)__-__GETNODEATTR($$CN,arch)__-statelite-compute
 check:rc==0
 cmd:lsdef $$CN |grep provmethod
 check:rc==0