From e976665d599fdd6c054b318323eca376d7295d8c Mon Sep 17 00:00:00 2001
From: Augustus <14297860+augustbleeds@users.noreply.github.com>
Date: Thu, 18 Dec 2025 15:55:13 -0500
Subject: [PATCH] Update reload.go

---
 framework/components/simple_node_set/reload.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/framework/components/simple_node_set/reload.go b/framework/components/simple_node_set/reload.go
index 25753a2c5..0343ad634 100644
--- a/framework/components/simple_node_set/reload.go
+++ b/framework/components/simple_node_set/reload.go
@@ -15,7 +15,11 @@ import (
 // UpgradeNodeSet updates nodes configuration TOML files
 // this API is discouraged, however, you can use it if nodes require restart or configuration updates, temporarily!
 func UpgradeNodeSet(t *testing.T, in *Input, bc *blockchain.Output, wait time.Duration) (*Output, error) {
-	uniq := fmt.Sprintf("%s-%s-%s", framework.DefaultCTFLogsDir, t.Name(), uuid.NewString()[0:4])
+	return UpgradeNodeSetWithTag(t.Name(), in, bc, wait)
+}
+
+func UpgradeNodeSetWithTag(tag string, in *Input, bc *blockchain.Output, wait time.Duration) (*Output, error) {
+	uniq := fmt.Sprintf("%s-%s-%s", framework.DefaultCTFLogsDir, tag, uuid.NewString()[0:4])
 	if _, err := framework.SaveContainerLogs(uniq); err != nil {
 		return nil, err
 	}