diff --git a/intellij/patches/bazel_intellij_aspect_filter.patch b/intellij/patches/bazel_intellij_aspect_filter.patch
index 53a8baf..01642cd 100644
--- a/intellij/patches/bazel_intellij_aspect_filter.patch
+++ b/intellij/patches/bazel_intellij_aspect_filter.patch
@@ -32,7 +32,7 @@
          "go_appengine_test",
      ]:
          sources = [f for src in getattr(ctx.rule.attr, "srcs", []) for f in src.files.to_list()]
-+        sources += [f for embed in getattr(ctx.rule.attr, "embed", []) for f in _collect_generated_go_sources(embed) or []]
++        sources += [f for embed in getattr(ctx.rule.attr, "embed", []) for f in _collect_generated_go_sources(embed, ctx, semantics) or []]
 +
          generated = [f for f in sources if not f.is_source]
      elif ctx.rule.kind == "go_wrap_cc":