diff --git a/platformio/builder/tools/pioide.py b/platformio/builder/tools/pioide.py
index 37cd55f0..7c3ec208 100644
--- a/platformio/builder/tools/pioide.py
+++ b/platformio/builder/tools/pioide.py
@@ -59,7 +59,10 @@ def _dump_includes(env):
             includes["toolchain"].extend([os.path.realpath(inc) for inc in glob(g)])
 
     includes["unity"] = []
-    unity_dir = get_core_package_dir("tool-unity", auto_install=False)
+    unity_dir = get_core_package_dir(
+        "tool-unity",
+        auto_install=os.path.isdir(env.GetProjectConfig().get_optional_dir("test")),
+    )
     if unity_dir:
         includes["unity"].append(unity_dir)
 
diff --git a/tests/commands/test_check.py b/tests/commands/test_check.py
index b8c8e65a..c631a613 100644
--- a/tests/commands/test_check.py
+++ b/tests/commands/test_check.py
@@ -154,7 +154,7 @@ def test_check_includes_passed(clirunner, check_dir):
             inc_count = l.count("-I")
 
     # at least 1 include path for default mode
-    assert inc_count > 1
+    assert inc_count > 0
 
 
 def test_check_silent_mode(clirunner, validate_cliresult, check_dir):