1 files changed, 6 insertions, 7 deletions
diff --git a/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotationArgumentProcessor.kt b/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotationArgumentProcessor.kt
index 3fd6634a..68e263c6 100644
--- a/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotationArgumentProcessor.kt
+++ b/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotationArgumentProcessor.kt
@@ -30,16 +30,15 @@ class AnnotationArgumentProcessor : AbstractTestProcessor() {
 
     override fun process(resolver: Resolver): List<KSAnnotated> {
         resolver.getSymbolsWithAnnotation("Bar", true).forEach {
-            val annotation = it.annotations.single()
-            annotation.arguments.forEach { it.accept(visitor, Unit) }
+            it.annotations.forEach { it.arguments.forEach { it.accept(visitor, Unit) } }
         }
 
-        val C = resolver.getClassDeclarationByName("C")!!
-        C.annotations.first().arguments.forEach { results.add(it.value.toString()) }
-        val ThrowsClass = resolver.getClassDeclarationByName("ThrowsClass")!!
-        ThrowsClass.declarations.filter {
+        val C = resolver.getClassDeclarationByName("C")
+        C?.annotations?.first()?.arguments?.forEach { results.add(it.value.toString()) }
+        val ThrowsClass = resolver.getClassDeclarationByName("ThrowsClass")
+        ThrowsClass?.declarations?.filter {
             it.simpleName.asString() == "throwsException"
-        }.forEach {
+        }?.forEach {
             it.annotations.single().annotationType.resolve().declaration.let {
                 results.add(it.toString())
             }