diff --git a/.github/workflows/python-app.yml b/.github/workflows/python-app.yml
index 2771c0e..3b33924 100644
--- a/.github/workflows/python-app.yml
+++ b/.github/workflows/python-app.yml
@@ -1,6 +1,3 @@
-# This workflow will install Python dependencies, run tests and lint with a single version of Python
-# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-python
-
 name: Build and test keras-mdn-layer
 
 on:
@@ -22,20 +19,30 @@ jobs:
 
     steps:
     - uses: actions/checkout@v4
+    - name: Install poetry
+      run: pipx install poetry
     - uses: actions/setup-python@v3
       with:
-        python-version: "3.11"
-    - uses: abatilo/actions-poetry@v2
+        python-version: '3.11'
+        cache: 'poetry'
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip
         poetry install
-    - name: Run tests.
-      run: |
-        poetry run pytest
     - name: Lint with flake8
       run: |
         # stop the build if there are Python syntax errors or undefined names
         poetry run flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
         # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
-        poetry run flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
\ No newline at end of file
+        poetry run flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
+    - name: Run tests.
+      run: |
+        poetry run coverage run --source=keras_mdn_layer -m pytest 
+    - name: Upload coverage.
+      run: poetry run coveralls
+      env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+    # - name: refresh coverage badge
+    #   uses: fjogeleit/http-request-action@v1
+    #   with:
+    #     url: https://camo.githubusercontent.com/46e89dfeb8bfbb9f0d13e8c5ed0d3779851bf0ecde89e303a3bf2a91bd956039/68747470733a2f2f636f766572616c6c732e696f2f7265706f732f6769746875622f63706d70657263757373696f6e2f696d70732f62616467652e7376673f6272616e63683d6d61696e
+    #     method: PURGE