diff --git a/tests/commands/test_init.py b/tests/commands/test_init.py
index 4310fc41..5e0172a7 100644
--- a/tests/commands/test_init.py
+++ b/tests/commands/test_init.py
@@ -16,10 +16,10 @@ import json
 from os import getcwd, makedirs
 from os.path import getsize, isdir, isfile, join
 
-from platformio import exception
 from platformio.commands.boards import cli as cmd_boards
 from platformio.commands.init import cli as cmd_init
 from platformio.project.config import ProjectConfig
+from platformio.project.exception import ProjectEnvsNotAvailableError
 
 
 def validate_pioproject(pioproject_dir):
@@ -59,7 +59,7 @@ def test_init_ide_without_board(clirunner, tmpdir):
     with tmpdir.as_cwd():
         result = clirunner.invoke(cmd_init, ["--ide", "atom"])
         assert result.exit_code != 0
-        assert isinstance(result.exception, exception.ProjectEnvsNotAvailable)
+        assert isinstance(result.exception, ProjectEnvsNotAvailableError)
 
 
 def test_init_ide_atom(clirunner, validate_cliresult, tmpdir):
diff --git a/tests/test_projectconf.py b/tests/test_projectconf.py
index 37193482..b601cece 100644
--- a/tests/test_projectconf.py
+++ b/tests/test_projectconf.py
@@ -16,8 +16,8 @@ import os
 
 import pytest
 
-from platformio.exception import UnknownEnvNames
 from platformio.project.config import ConfigParser, ProjectConfig
+from platformio.project.exception import UnknownEnvNamesError
 
 BASE_CONFIG = """
 [platformio]
@@ -115,7 +115,7 @@ def test_warnings(config):
     assert len(config.warnings) == 2
     assert "lib_install" in config.warnings[1]
 
-    with pytest.raises(UnknownEnvNames):
+    with pytest.raises(UnknownEnvNamesError):
         config.validate(["non-existing-env"])