diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index 2f80821..3525965 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -35,7 +35,7 @@ jobs:
       - name: Build wheels in docker
         shell: bash
         run: |
-          docker run --rm -v `pwd`:/project -w /project quay.io/pypa/manylinux2010_x86_64 bash .github/workflows/Docker/buildwheel.sh
+          docker run --rm -v `pwd`:/project -w /project quay.io/pypa/manylinux2014_x86_64:latest bash .github/workflows/Docker/buildwheel.sh
       - name: Upload Wheels
         uses: actions/upload-artifact@v2
         with:
diff --git a/diploshic/diploSHIC b/diploshic/diploSHIC
index f0134bc..901bc46 100644
--- a/diploshic/diploSHIC
+++ b/diploshic/diploSHIC
@@ -534,7 +534,7 @@ if argsDict["mode"] == "train":
     print("diploSHIC loss: %f" % score[0])
     print("diploSHIC accuracy: %f" % score[1])
     if confusionFile:
-        from misc import plot_confusion_matrix
+        from diploshic.misc import plot_confusion_matrix
         import matplotlib.pyplot as plt
 
         plot_confusion_matrix(
diff --git a/setup.py b/setup.py
index dae6c2d..167bf71 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
                                 "diploshic/utils.c"],
                       )
 setup(name='diploSHIC',
-      version='1.0.4',
+      version='1.0.5',
       description='diploSHIC',
       long_description=long_description,
       long_description_content_type="text/markdown",