diff --git a/setup.cfg b/setup.cfg
index 2027b53..0e37183 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -37,7 +37,7 @@ install_requires =
     matplotlib-inline
     pexpect>4.3; sys_platform != "win32"
     pickleshare
-    prompt_toolkit>=2.0.0,<3.1.0,!=3.0.0,!=3.0.1
+    prompt_toolkit>3.0.1,<3.1.0
     pygments>=2.4.0
     setuptools>=18.5
     stack_data