Move from after_script to after_success