diff --git a/tests/test_elements_worker/test_training.py b/tests/test_elements_worker/test_training.py
index 9b6f014a8fc19c1a5bd5ba13425f3cbb5fbb82d6..73970081a9c459af838eee7141dd8e6cc52c0721 100644
--- a/tests/test_elements_worker/test_training.py
+++ b/tests/test_elements_worker/test_training.py
@@ -56,6 +56,8 @@ def test_create_model_version(tag, description):
     model_hash = "hash"
     archive_hash = "archive_hash"
     size = "30"
+    tag = "tag"
+    description = "description"
     model_version_details = {
         "id": model_version_id,
         "model_id": model_id,