diff --git a/src/test/java/org/openrewrite/java/migrate/jspecify/MigrateToJspecifyTest.java b/src/test/java/org/openrewrite/java/migrate/jspecify/MigrateToJspecifyTest.java
index 4b094becd..be2a78d9c 100644
--- a/src/test/java/org/openrewrite/java/migrate/jspecify/MigrateToJspecifyTest.java
+++ b/src/test/java/org/openrewrite/java/migrate/jspecify/MigrateToJspecifyTest.java
@@ -84,7 +84,7 @@ class Bar {
                   }
                   """
               ),
-              // package-info.java
+              //language=java
               java(
                 """
                   @ParametersAreNonnullByDefault
@@ -96,8 +96,9 @@ class Bar {
                   @NullMarked
                   package org.openrewrite.example;
 
-                  import org.jspecify.annotation.NullMarked;
-                  """
+                  import org.jspecify.annotations.NullMarked;
+                  """,
+                spec -> spec.path("src/main/java/org/openrewrite/example/package-info.java")
               )
             ),
             //language=xml