diff --git a/tensorflow_probability/substrates/meta/rewrite.py b/tensorflow_probability/substrates/meta/rewrite.py
index <HASH>..<HASH> 100644
--- a/tensorflow_probability/substrates/meta/rewrite.py
+++ b/tensorflow_probability/substrates/meta/rewrite.py
@@ -260,16 +260,17 @@ def main(argv):
 
   substrate = 'jax' if FLAGS.numpy_to_jax else 'numpy'
   path = filename.split('/python/')[1]
-  header = '\n'.join([
+  footer = '\n'.join([
+      '\n',
       '# ' + '@' * 78,
       '# This file is auto-generated by substrates/meta/rewrite.py',
       '# It will be surfaced by the build system as a symlink at:',
       '#   `tensorflow_probability/substrates/{}/{}`'.format(substrate, path),
       '# For more info, see substrate_runfiles_symlinks in build_defs.bzl',
       '# ' + '@' * 78,
-      '\n# (This notice adds 10 to line numbering.)\n\n\n'])
+      ])
 
-  print(header + contents, file=open(1, 'w', encoding='utf-8', closefd=False))
+  print(contents + footer, file=open(1, 'w', encoding='utf-8', closefd=False))
 
 
 if __name__ == '__main__':
Move the JAX/NumPy generated files notice to the end.

This is to make the line numbers for errors correspond to the original sources.

PiperOrigin-RevId: <I>