Fix after_run