diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index c35d285..07c2344 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -106,10 +106,6 @@ jobs:
         uses: actions/setup-python@v2
         with:
           python-version: ${{ matrix.python-version }}
-      - name: Upgrade setuptools
-        run: pip install --upgrade setuptools
-      - name: Install wheel  # wheel is the default package of local environment but docker environment misses the package so we manually install it
-        run: pip install wheel
       - name: Install PyTorch
         run: pip install torch==${{matrix.torch}}+cpu torchvision==${{matrix.torchvision}}+cpu -f https://download.pytorch.org/whl/torch_stable.html
       - name: Build and install
@@ -118,9 +114,10 @@ jobs:
         run: python -c "import mim"
       - name: Install unittest dependencies
         run: pip install -r requirements/tests.txt
-      - name: Run unittests and generate coverage report
+      - name: Run unittests
+        run: coverage run --branch --source=mim -m pytest tests/
+      - name: Generate coverage report
         run: |
-          coverage run --branch --source=mim -m pytest tests/
           coverage xml
           coverage report -m
       - name: Upload coverage to Codecov