diff --git a/tests/test_main.py b/tests/test_main.py
index f7e05a9..9c47dbe 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -6,40 +6,32 @@
 
 
 @pytest.fixture
-def app():
+def app() -> Flask:
     app = Flask(__name__)
     app.config["TESTING"] = True
     return app
 
 
 @pytest.fixture
-def runner(app):
+def runner(app: Flask) -> CliRunner:
     return CliRunner()
 
 
-def test_ipython_shell(runner, app):
+def test_ipython_shell(runner: CliRunner, app: Flask) -> None:
     with app.app_context():
         result = runner.invoke(shell, ["--shelltype", "ipython"])
     assert result.exit_code == 0
     assert "IPython" in result.output
 
 
-# def test_bpython_shell(runner, app):
-#     with app.app_context():
-#         result = runner.invoke(shell, ["--shelltype", "bpython"])
-#         assert result.output == ""
-#     assert result.exit_code == 0
-#     assert "BPython" in result.output
-
-
-def test_pypython_shell(runner, app):
+def test_ptpython_shell(runner: CliRunner, app: Flask) -> None:
     with app.app_context():
         result = runner.invoke(shell, ["--shelltype", "ptpython"])
     assert result.exit_code == 0
     assert "PTPython" in result.output
 
 
-def test_python_shell(runner, app):
+def test_python_shell(runner: CliRunner, app: Flask) -> None:
     with app.app_context():
         result = runner.invoke(shell, ["--shelltype", "python"])
     assert result.exit_code == 0