diff --git a/Makefile b/Makefile
index 1c7aeb77..3025537b 100644
--- a/Makefile
+++ b/Makefile
@@ -41,12 +41,11 @@ clean:
 	find . -type f -name "*.pth" -exec rm -f {} \; || true
 	find . -type f -name "mlflow.db" -exec rm -f {} \; || true
 	find . -type d -name "mlruns" -exec rm -rf {} \; 2>/dev/null || true
-	find . -type d -name "mlruns" -exec rm -rf {} \; 2>/dev/null || true
 	find . -type d -name "__pycache__" -exec rm -rf {} \;  2>/dev/null || true
 	find . -type d -name "*tfevents*" -exec rm -rf {} \;  2>/dev/null || true
 	find . -type d -name "*output_data*" -exec rm -rf {} \;  2>/dev/null || true
 	find . -type f -name "*nohup*" -exec rm -rf {} \;  2>/dev/null || true
-	# sphinx-build -M clean docs docs/_build This needs to be fixed.
+	sphinx-build -M clean docs docs/_build This needs to be fixed.
 
 # Build the HTML documentation using Sphinx
 .PHONY: docs
diff --git a/examples/llm_complex/llm_test_runner.py b/examples/llm_complex/llm_test_runner.py
index 584c7b4f..16e5b173 100644
--- a/examples/llm_complex/llm_test_runner.py
+++ b/examples/llm_complex/llm_test_runner.py
@@ -65,7 +65,7 @@ def all():
         write_log(f"\n\nStarting for combination {i}/{len(combinations)}: {config}")
 
         try:
-            success = run_test(max_runs=5, config=config)
+            success = run_test(max_runs=3, config=config)
             if not success:
                 write_log(str(i) + "--> " + str(config))
                 sys.exit(1)
diff --git a/pyproject.toml b/pyproject.toml
index 330b3f07..02880a7a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,7 +11,7 @@ dependencies = [
     "msgpack",
     "omegaconf",
     "pandas",
-    "psutil",
+    "psutil>=6.1.1",
     "py-cpuinfo",
     "redis",
     "requests",
@@ -64,13 +64,13 @@ mlflow = ["mlflow-skinny", "SQLAlchemy", "alembic", "watchdog"]
 nvidia = ["nvidia-ml-py"]
 tensorboard = ["tensorboard", "tensorflow", "tbparse"]
 dev = [
+    "flowcept[docs]",
     "jupyterlab",
     "nbmake",
     "pika",
     "pytest",
     "ruff",
     "pyyaml",
-    "psutil>=6.1.1"
 ]
 # Torch and some other ml-specific libs, only used for dev purposes, require the following specific versions.
 ml_dev = [