diff --git a/main.py b/main.py
index 4003839..679ce44 100644
--- a/main.py
+++ b/main.py
@@ -18,10 +18,10 @@ console = Console()
 # --- Safe-global engedélyezés TorchVersion számára (checkpoint hack) ---
 from torch.serialization import add_safe_globals
 from torch.torch_version import TorchVersion
-from pyannote.audio.core.task import Specifications
+from pyannote.audio.core.task import Specifications, Problem
 
 # allowloading checkpoints that store these classes (PyTorch 2.6+ weights_only change)
-add_safe_globals([TorchVersion, Specifications])
+add_safe_globals([TorchVersion, Specifications, Problem])
 
 # --- FFmpeg és Whisper a saját rendszered pipeline-jához, HF Token kötelező ---
 if len(sys.argv) < 2: