diff --git a/conda.yml b/conda.yml
index a1363a240..87e438489 100644
--- a/conda.yml
+++ b/conda.yml
@@ -3,10 +3,9 @@ channels:
   - conda-forge
   - defaults
 dependencies:
-  - python=3.10
-  - pyyaml
-  - hydra-core=1.3.2
-  - pip=23.3.1
+  - pyyaml=5.3.1
+  - hydra-core=1.0.6
+  - pip=20.3.3
+  - mlflow=1.14.1
   - pip:
-      - mlflow==2.8.1
       - wandb==0.16.0
diff --git a/environment.yml b/environment.yml
index 2ab326c1a..00763b169 100644
--- a/environment.yml
+++ b/environment.yml
@@ -3,12 +3,14 @@ channels:
   - conda-forge
   - defaults
 dependencies:
-  - python=3.10
+  - mlflow==1.14.1
   - hydra-core=1.3.2
+  - ipython=7.21.0
+  - notebook=6.2.0
   - matplotlib=3.8.2
+  - cookiecutter=1.7.2
   - pandas=2.1.3
-  - jupyterlab=4.0.9
+  - jupyterlab=3.0.10
   - pip=23.3.1
   - pip:
-      - mlflow==2.8.1
       - wandb==0.16.0
\ No newline at end of file
diff --git a/main.py b/main.py
index d91bc1e67..c313d6f45 100644
--- a/main.py
+++ b/main.py
@@ -115,7 +115,7 @@ def go(config: DictConfig):
         if "test_regression_model" in active_steps:
 
             _ = mlflow.run(
-                f"{config['main']['components_repository']}/test_regression_model",
+                os.path.join(hydra.utils.get_original_cwd(), "components", "test_regression_model"),
                 "main",
                 parameters={
                     "mlflow_model": "random_forest_export:prod",