diff --git a/gazelle/python/testdata/remove_invalid_binary/BUILD.in b/gazelle/python/testdata/remove_invalid_binary/BUILD.in
index 87d357139b..f4bfd65c1b 100644
--- a/gazelle/python/testdata/remove_invalid_binary/BUILD.in
+++ b/gazelle/python/testdata/remove_invalid_binary/BUILD.in
@@ -1,11 +1,13 @@
 load("@rules_python//python:defs.bzl", "py_binary", "py_library")
 
 py_library(
-    name = "keep_library",
+    name = "remove_invalid_binary",
+    srcs = ["__init__.py"],
     deps = ["//keep_binary:foo"],
 )
+
 py_binary(
-    name = "remove_invalid_binary",
+    name = "remove_invalid_binary_bin",
     srcs = ["__main__.py"],
     data = ["testdata/test.txt"],
     visibility = ["//:__subpackages__"],
diff --git a/gazelle/python/testdata/remove_invalid_binary/BUILD.out b/gazelle/python/testdata/remove_invalid_binary/BUILD.out
index 069188f5ca..a217b4bdaf 100644
--- a/gazelle/python/testdata/remove_invalid_binary/BUILD.out
+++ b/gazelle/python/testdata/remove_invalid_binary/BUILD.out
@@ -1,6 +1,7 @@
 load("@rules_python//python:defs.bzl", "py_library")
 
 py_library(
-    name = "keep_library",
-    deps = ["//keep_binary:foo"],
+    name = "remove_invalid_binary",
+    srcs = ["__init__.py"],
+    visibility = ["//:__subpackages__"],
 )
diff --git a/gazelle/python/testdata/remove_invalid_binary/__init__.py b/gazelle/python/testdata/remove_invalid_binary/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
diff --git a/gazelle/python/testdata/remove_invalid_library/BUILD.in b/gazelle/python/testdata/remove_invalid_library/BUILD.in
index 3f24c8df35..815e673ee1 100644
--- a/gazelle/python/testdata/remove_invalid_library/BUILD.in
+++ b/gazelle/python/testdata/remove_invalid_library/BUILD.in
@@ -1,4 +1,4 @@
-load("@rules_python//python:defs.bzl", "py_library")
+load("@rules_python//python:defs.bzl", "py_library", "py_test")
 
 py_library(
     name = "remove_invalid_library",
@@ -14,3 +14,8 @@ py_library(
         "@pypi//foo",
     ],
 )
+
+py_test(
+    name = "my_test",
+    srcs = ["my_test.py"],
+)
diff --git a/gazelle/python/testdata/remove_invalid_library/BUILD.out b/gazelle/python/testdata/remove_invalid_library/BUILD.out
index 4a6fffa183..80d47076c3 100644
--- a/gazelle/python/testdata/remove_invalid_library/BUILD.out
+++ b/gazelle/python/testdata/remove_invalid_library/BUILD.out
@@ -1,4 +1,4 @@
-load("@rules_python//python:defs.bzl", "py_library")
+load("@rules_python//python:defs.bzl", "py_library", "py_test")
 
 py_library(
     name = "deps_with_no_srcs_library",
@@ -8,3 +8,8 @@ py_library(
         "@pypi//foo",
     ],
 )
+
+py_test(
+    name = "my_test",
+    srcs = ["my_test.py"],
+)
diff --git a/gazelle/python/testdata/remove_invalid_library/my_test.py b/gazelle/python/testdata/remove_invalid_library/my_test.py
new file mode 100644
index 0000000000..e69de29bb2