diff --git a/.github/workflows/dependency-testing.yaml b/.github/workflows/dependency-testing.yaml
index b7deb2e35..e8714603e 100644
--- a/.github/workflows/dependency-testing.yaml
+++ b/.github/workflows/dependency-testing.yaml
@@ -48,6 +48,9 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
+      - name: Free Disk Space
+        run: ./disk_clean.sh
+
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
         with:
diff --git a/pyproject.toml b/pyproject.toml
index 9e2e6fbbd..0a2b3a989 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "validmind"
-version = "2.10.3"
+version = "2.10.4"
 description = "ValidMind Library"
 readme = "README.pypi.md"
 requires-python = ">=3.9,<3.13"
diff --git a/validmind/__version__.py b/validmind/__version__.py
index 21e33ba2c..e3b571e0f 100644
--- a/validmind/__version__.py
+++ b/validmind/__version__.py
@@ -1 +1 @@
-__version__ = "2.10.3"
+__version__ = "2.10.4"
diff --git a/validmind/ai/test_descriptions.py b/validmind/ai/test_descriptions.py
index 617843c40..c26870239 100644
--- a/validmind/ai/test_descriptions.py
+++ b/validmind/ai/test_descriptions.py
@@ -121,9 +121,7 @@ def generate_description(
             "test_description": test_description,
             "title": title,
             "summary": _truncate_summary(summary, test_id),
-            "figures": [
-                figure._get_b64_url() for figure in ([] if tables else figures)
-            ],
+            "figures": [figure._get_b64_url() for figure in figures or []],
             "additional_context": additional_context,
             "instructions": instructions,
         }