diff --git a/main.py b/main.py
index 7fc7a55f8c468008d409f297b5321350a74acdc4..5622ea92b41d32093802446086750ceba59f8a32 100644
--- a/main.py
+++ b/main.py
@@ -3,7 +3,7 @@ import sys
 
 from termcolor import colored
 from src.drive_handle import DriveHandle
-from test_orchestrator import TestOrchestrator
+from src.test_orchestrator import TestOrchestrator
 
 
 if __name__ == "__main__":
diff --git a/src/drive_handle.py b/src/drive_handle.py
index c979163bb9d67650e453300f870feb4f762c5cf4..8862f4f91283eb6e02f9d91eaac11eee83b500de 100644
--- a/src/drive_handle.py
+++ b/src/drive_handle.py
@@ -154,7 +154,7 @@ class DriveHandle:
         pattern_symbol: bytes = b'A' * 512
         progressbar = Spinner('[DriveHandle] Writing the pattern')
 
-        volume_path = subprocess.check_output(['powershell.exe', f'(Get-Volume -DriveLetter {self._drive_letter}).UniqueID'])
+        volume_path = subprocess.check_output(['powershell.exe', f'(Get-Volume -DriveLetter {self._drive_letter}).UniqueID']).decode("utf-8").replace("\r\n", "")
 
         raw_disk = open(volume_path, 'wb')
         try: