diff --git a/setup.py b/setup.py
index 156cd1a..2a9b81e 100755
--- a/setup.py
+++ b/setup.py
@@ -19,7 +19,7 @@ description = 'CLI for MySQL Database. With auto-completion and syntax highlight
 install_requirements = [
     'click >= 7.0',
     'Pygments >= 1.6',
-    'prompt_toolkit>=3.0.6,<4.0.0',
+    'prompt_toolkit>=2.0.6,<3.0.0',
     'PyMySQL >= 0.9.2',
     'sqlparse>=0.3.0,<0.4.0',
     'configobj >= 5.0.5',
diff --git a/test/features/environment.py b/test/features/environment.py
index 1a49dbe..422c264 100644
--- a/test/features/environment.py
+++ b/test/features/environment.py
@@ -17,7 +17,6 @@ def before_all(context):
     os.environ['COLUMNS'] = "100"
     os.environ['EDITOR'] = 'ex'
     os.environ['LC_ALL'] = 'en_US.utf8'
-    os.environ['PROMPT_TOOLKIT_NO_CPR'] = '1'
 
     test_dir = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
     login_path_file = os.path.join(test_dir, 'mylogin.cnf')