diff --git a/src/argilla_llama_index/llama_index_handler.py b/src/argilla_llama_index/llama_index_handler.py
index 623f5c8..3cc8168 100644
--- a/src/argilla_llama_index/llama_index_handler.py
+++ b/src/argilla_llama_index/llama_index_handler.py
@@ -124,7 +124,7 @@ def __init__(
             api_key = DEFAULT_API_KEY
 
         
-        rg.init(api_key=api_key, api_url=api_url)
+        rg.init(api_key=api_key, api_url=api_url, workspace=workspace_name or rg.get_workspace())
 
         # Set the Argilla variables
         self.dataset_name = dataset_name