diff --git a/WAS_Node_Suite.py b/WAS_Node_Suite.py
index 0db7094..ed4a3bc 100644
--- a/WAS_Node_Suite.py
+++ b/WAS_Node_Suite.py
@@ -831,8 +831,8 @@ class TextTokens:
         self.tokens = {
             '[time]': str(time.time()).replace('.','_'),
             '[hostname]': socket.gethostname(),
-            '[cuda_device]': comfy.model_management.get_torch_device(),
-            '[cuda_name]': comfy.model_management.get_torch_device_name(),
+            '[cuda_device]': str(comfy.model_management.get_torch_device()),
+            '[cuda_name]': str(comfy.model_management.get_torch_device_name(device=comfy.model_management.get_torch_device())),
         }
 
         if '.' in self.tokens['[time]']: