Make sure that output files are written out at the final step. Should fix issues with restarting reported by @PabloPiaggi