diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 5771d488..a2122af0 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -10,26 +10,7 @@ on:
 
 jobs:
   codestyle:
-    runs-on: ubuntu-latest
-    if: ${{ github.event_name == 'push' || !github.event.pull_request.draft }}
-    steps:
-    - uses: actions/checkout@v4
-    - name: Set up Python 3.10
-      uses: actions/setup-python@v5
-      with:
-        python-version: "3.10"
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        python -m pip install black flake8
-    - name: Code Style (Black/Flake8)
-      run: |
-        # Black code style
-        black --check --diff pytorch_widedeep tests examples setup.py
-        # Stop the build if there are Python syntax errors or undefined names
-        flake8 . --count --select=E901,E999,F821,F822,F823 --ignore=E266 --show-source --statistics
-        # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
-        flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --ignore=E203,E266,E501,E721,E722,F401,F403,F405,F811,W503,C901 --statistics
+    # ... (keep the codestyle job as is)
 
   test:
     runs-on: ubuntu-latest
@@ -37,7 +18,7 @@ jobs:
     strategy:
       fail-fast: true
       matrix:
-        python-version: ["3.10"]
+        python-version: [3.10"]
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python ${{ matrix.python-version }}
@@ -49,17 +30,23 @@ jobs:
         python -m pip install --upgrade pip
         python -m pip install pytest-cov codecov faker
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+    - name: Set coverage directory
+      run: echo "COVERAGE_DIR=./coverage_data" >> $GITHUB_ENV
     - name: Test with pytest
       run: |
-        pytest --doctest-modules pytorch_widedeep/models --cov-report xml --cov-report term --disable-pytest-warnings --cov=pytorch_widedeep/models tests/test_model_functioning
+        mkdir -p $COVERAGE_DIR
+        pytest --doctest-modules pytorch_widedeep/models --cov-report xml:$COVERAGE_DIR/coverage.xml --cov-report term --disable-pytest-warnings --cov=pytorch_widedeep/models tests/test_model_functioning --cov-config=.coveragerc
     - name: Debug - List files
       run: |
+        echo "Contents of the current directory:"
         ls -la
+        echo "Contents of the coverage directory:"
+        ls -la $COVERAGE_DIR
     - name: Upload coverage
       uses: actions/upload-artifact@v4
       with:
         name: coverage-${{ matrix.python-version }}
-        path: .coverage
+        path: ${{ env.COVERAGE_DIR }}
         if-no-files-found: error
 
   finish:
@@ -82,12 +69,13 @@ jobs:
         path: coverage-files
     - name: Debug - List downloaded files
       run: |
+        echo "Contents of the coverage-files directory:"
         ls -R coverage-files
     - name: Convert coverage
       run: |
         coverage combine coverage-files/*/.coverage
-        coverage report --fail-under=90
         coverage xml
+        coverage report --fail-under=90
     - name: upload coverage to Codecov
       uses: codecov/codecov-action@v4
       with: