diff --git a/tooling/impl/src/main/java/com/itsaky/androidide/tooling/impl/sync/AndroidProjectModelBuilder.kt b/tooling/impl/src/main/java/com/itsaky/androidide/tooling/impl/sync/AndroidProjectModelBuilder.kt
index 4ec860e99..5799479ed 100644
--- a/tooling/impl/src/main/java/com/itsaky/androidide/tooling/impl/sync/AndroidProjectModelBuilder.kt
+++ b/tooling/impl/src/main/java/com/itsaky/androidide/tooling/impl/sync/AndroidProjectModelBuilder.kt
@@ -70,6 +70,8 @@ class AndroidProjectModelBuilder(initializationParams: InitializeProjectParams)
       it.dontBuildAndroidTestRuntimeClasspath = true
       it.dontBuildTestFixtureRuntimeClasspath = true
       it.dontBuildUnitTestRuntimeClasspath = true
+      it.dontBuildHostTestRuntimeClasspath = emptyMap()
+      it.dontBuildScreenshotTestRuntimeClasspath = true
     }
 
     controller.findModel(module, ProjectSyncIssues::class.java)?.also { syncIssues ->
diff --git a/tooling/impl/src/test/java/com/itsaky/androidide/tooling/impl/CIOnlyToolingApiTests.kt b/tooling/impl/src/test/java/com/itsaky/androidide/tooling/impl/CIOnlyToolingApiTests.kt
index a8d0e28b0..14d211277 100644
--- a/tooling/impl/src/test/java/com/itsaky/androidide/tooling/impl/CIOnlyToolingApiTests.kt
+++ b/tooling/impl/src/test/java/com/itsaky/androidide/tooling/impl/CIOnlyToolingApiTests.kt
@@ -41,7 +41,7 @@ class CIOnlyToolingApiTests {
 
   /**
    * Tests the functionality of the tooling API implementation against multiple versions of the
-   * Android Gradle Plugin. This test runs only in the CI environment.
+   * Android Gradle   Plugin. This test runs only in the CI environment.
    */
   @Test
   fun `test CI-only simple multi module project initialization with multiple AGP versions`() {