diff --git a/.travis.yml b/.travis.yml
index c238f5c..1e842be 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -5,10 +5,10 @@ notifications:
 cache: pip
 
 python:
-  - "3.6"
+  - "3.8"
 
 env:
-    - TRAVISCI=true PYTHON=3.6
+    - TRAVISCI=true PYTHON=3.8
 
 services:
   - xvfb
diff --git a/requirements.txt b/requirements.txt
index 4d153cb..86555f8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 numpy
-pyyaml==5.3.1 # Workaround for networkx issue 4548
+pyyaml
 networkx>=2.1
 matplotlib
 pytools
diff --git a/setup.py b/setup.py
index 50f4401..7090120 100644
--- a/setup.py
+++ b/setup.py
@@ -57,9 +57,10 @@ def license():
       keywords=['graph', 'calibration framework'],
       url='https://gitlab.com/AdriaanRol/AutoDepGraph',
       classifiers=['Development Status :: 4 - Beta', 'Intended Audience :: Science/Research',
-                   'Programming Language :: Python :: 3.6',
                    'Programming Language :: Python :: 3.7',
                    'Programming Language :: Python :: 3.8',
+                   'Programming Language :: Python :: 3.9',
+                   'Programming Language :: Python :: 3.10',
                    'License :: OSI Approved :: MIT License',
                    ]
       )