diff --git a/setup.py b/setup.py
index bd1939925f..f379bd782a 100644
--- a/setup.py
+++ b/setup.py
@@ -29,7 +29,7 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "transformers>=4.36,<4.45",
-    "optimum@git+https://github.com/huggingface/optimum.git",
+    "optimum~=1.22",
     "datasets>=1.4.0",
     "sentencepiece",
     "setuptools",
diff --git a/tests/openvino/test_training.py b/tests/openvino/test_training.py
index 1f88b501ae..014b802688 100644
--- a/tests/openvino/test_training.py
+++ b/tests/openvino/test_training.py
@@ -805,7 +805,7 @@ def test_training(self, _, desc: OVTrainerTestDescriptor):
         self.run_ovtrainer_training_checks(desc)
 
     def prepare_model_and_dataset(self, desc: OVTrainerTestDescriptor):
-        self.dataset = load_dataset("anton-l/superb_dummy", "ks")
+        self.dataset = load_dataset("anton-l/superb_dummy", "ks", trust_remote_code=True)
         self.num_labels = len(self.dataset["test"].features["label"].names)
 
         self.feature_extractor = AutoFeatureExtractor.from_pretrained(desc.model_id)