diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index ce50c2b5..a6991950 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -3,21 +3,20 @@ repos:
     rev: 20.8b1
     hooks:
     - id: black
-      language_version: python3.8
+      language_version: python3
 -   repo: https://gitlab.com/pycqa/flake8
-    rev: 3.8.3
+    rev: 3.8.4
     hooks:
     - id: flake8
 -   repo: https://github.com/pre-commit/mirrors-mypy
-    rev: 'v0.790'
+    rev: 'v0.812'
     hooks:
     - id: mypy
       args: [--no-strict-optional, --ignore-missing-imports]
       additional_dependencies: [cma==3.0.3,
-                                matplotlib==3.3.2,
-                                numpy==1.18.5, 
+                                numpy==1.19.5, 
                                 scipy==1.5.2,
-                                tensorflow==2.3.1, 
-                                tensorflow-probability==0.11.1, 
-                                tensorflow-estimator==2.3.0
+                                tensorflow==2.4.1, 
+                                tensorflow-probability==0.12.1, 
+                                tensorflow-estimator==2.4.0
                                 ]
diff --git a/README.md b/README.md
index fc0578ec..b2f58413 100755
--- a/README.md
+++ b/README.md
@@ -3,6 +3,7 @@
 [![codecov](https://codecov.io/gh/q-optimize/c3/branch/master/graph/badge.svg)](https://codecov.io/gh/q-optimize/c3)
 [![Language grade: Python](https://img.shields.io/lgtm/grade/python/g/q-optimize/c3.svg?logo=lgtm&logoWidth=18)](https://lgtm.com/projects/g/q-optimize/c3/context:python)
 [![Documentation Status](https://readthedocs.org/projects/c3-toolset/badge/?version=latest)](https://c3-toolset.readthedocs.io/en/latest/?badge=latest)
+[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![PyPI version fury.io](https://badge.fury.io/py/c3-toolset.svg)](https://pypi.python.org/pypi/c3-toolset/)
 [![PyPI license](https://img.shields.io/pypi/l/c3-toolset.svg)](https://pypi.python.org/pypi/c3-toolset/)
 [![PyPI pyversions](https://img.shields.io/pypi/pyversions/c3-toolset.svg)](https://pypi.python.org/pypi/c3-toolset/)
diff --git a/requirements.txt b/requirements.txt
index 5d7a26ca..cfbc076d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,8 +15,7 @@ scipy==1.5.2
 six==1.15.0
 Sphinx==3.2.1
 sphinx-autoapi==1.4.0
-tensorboard==2.4.0
-tensorflow==2.4.0
+tensorflow==2.4.1
 tensorflow-estimator==2.4.0
 tensorflow-probability==0.12.1
 xenon==0.7.1
diff --git a/setup.py b/setup.py
index c313007a..fe024d54 100644
--- a/setup.py
+++ b/setup.py
@@ -42,7 +42,7 @@
         "rich==9.2.0",
         "numpy==1.19.5",
         "scipy==1.5.2",
-        "tensorflow==2.4.0",
+        "tensorflow==2.4.1",
         "tensorflow-estimator==2.4.0",
         "tensorflow-probability==0.12.1",
     ],