diff --git a/llmebench/models/Petals.py b/llmebench/models/Petals.py
index b25965e0..e0f6711d 100644
--- a/llmebench/models/Petals.py
+++ b/llmebench/models/Petals.py
@@ -127,6 +127,6 @@ def prompt(self, processed_input):
                 if not response["ok"]:
                     raise PetalsFailure("processing", response["traceback"])
             else:
-                raise PetalsFailure(connect_message["traceback"])
+                raise PetalsFailure("connection", connect_message["traceback"])
 
         return response