diff --git a/mz_train_tools_core.py b/mz_train_tools_core.py
index 18d2ceb..f7c1975 100644
--- a/mz_train_tools_core.py
+++ b/mz_train_tools_core.py
@@ -33,9 +33,9 @@ def MZ_KohyaSSInitWorkspace_call(args={}):
         result = subprocess.run(
             ["git", "branch"], cwd=kohya_ss_lora_dir, stdout=subprocess.PIPE, check=True)
 
-        print(f"current branch: {result.stdout}")
+        print(f"current branch: {result.stdout.decode()}")
 
-        if branch.encode() not in result.stdout:
+        if branch not in result.stdout.decode():
             subprocess.run(
                 ["git", "remote", "set-branches", "origin", branch], cwd=kohya_ss_lora_dir, check=True)
             subprocess.run(