fix types

abbbi · Nov 5, 2024 · 0a36219 · 0a36219
1 parent b099a16
commit 0a36219
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/libvirtnbdbackup/restore/disk.py b/libvirtnbdbackup/restore/disk.py
@@ -118,6 +118,6 @@ def restore(  # pylint: disable=too-many-branches
 
     if args.adjust_config is True:
         restConfig = vmconfig.removeUuid(restConfig.decode())
-        restConfig = vmconfig.setVMName(args, restConfig)
+        restConfig = vmconfig.setVMName(args, restConfig.decode())
 
     return restConfig
diff --git a/libvirtnbdbackup/restore/vmconfig.py b/libvirtnbdbackup/restore/vmconfig.py
@@ -54,7 +54,6 @@ def removeUuid(vmConfig: str) -> bytes:
     """Remove the auto generated UUID from the config file to allow
     for restore into new name"""
     tree = xml.asTree(vmConfig)
-
     try:
         logging.info("Removing uuid setting from vm config.")
         uuid = tree.xpath("uuid")[0]
@@ -65,7 +64,7 @@ def removeUuid(vmConfig: str) -> bytes:
     return xml.ElementTree.tostring(tree, encoding="utf8", method="xml")
 
 
-def setVMName(args: Namespace, vmConfig: bytes) -> bytes:
+def setVMName(args: Namespace, vmConfig: str) -> bytes:
     """Change / set the VM name to be restored"""
     tree = xml.asTree(vmConfig)
     name = tree.xpath("name")[0]