diff --git a/wscript b/wscript
index 0ea9c6050..e60bc7fce 100644
--- a/wscript
+++ b/wscript
@@ -544,6 +544,9 @@ def build(bld):
         # want to do is run a test program.
         if not Options.options.compile_targets:
             Options.options.compile_targets = os.path.basename(program_name)
+            for gen in bld.all_task_gen:
+                if type(gen).__name__ in ['ns3header_taskgen', 'ns3moduleheader_taskgen']:
+                    gen.post()
 
     if Options.options.regression or Options.options.regression_generate:
         regression_traces = env['REGRESSION_TRACES']