diff --git a/allegro.py b/allegro.py
index 9e996c4..33c3b8d 100644
--- a/allegro.py
+++ b/allegro.py
@@ -4,7 +4,7 @@ import weaviate
 import numpy as np
 from datasets import Dataset
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TrainingArguments, Trainer, DataCollatorForSeq2Seq
-from weaviate.classes.config import ConnectionParams
+from weaviate.connect import ConnectionParams
 import weaviate
 
 # 1️⃣ Połączenie z Weaviate