Change after_script to after_success