Change the way post_process is called