diff --git a/tests/unit/steps/argilla/test_preference.py b/tests/unit/steps/argilla/test_preference.py
index 28a7d5008c..5ff95782b0 100644
--- a/tests/unit/steps/argilla/test_preference.py
+++ b/tests/unit/steps/argilla/test_preference.py
@@ -23,7 +23,6 @@
 def mock_dataset() -> rg.Dataset:  # type: ignore
     return rg.Dataset(
         name="dataset",
-        workspace=rg.Workspace("workspace"),  # type: ignore
         settings=rg.Settings(
             fields=[
                 rg.TextField(name="id", title="id"),  # type: ignore
diff --git a/tests/unit/steps/argilla/test_text_generation.py b/tests/unit/steps/argilla/test_text_generation.py
index 9e20b33efc..5679c54517 100644
--- a/tests/unit/steps/argilla/test_text_generation.py
+++ b/tests/unit/steps/argilla/test_text_generation.py
@@ -21,7 +21,6 @@
 
 MockDataset = rg.Dataset(
     name="dataset",
-    workspace=rg.Workspace("workspace"),  # type: ignore
     settings=rg.Settings(
         fields=[
             rg.TextField(name="id", title="id"),  # type: ignore