diff --git a/tox.ini b/tox.ini
index 83bfe822..9c369b58 100644
--- a/tox.ini
+++ b/tox.ini
@@ -16,5 +16,6 @@ deps =
     pytest-cov
     pytest-mock
     pytest-timeout
+extras = tensorflow
 passenv =
     NUMBA_DISABLE_JIT