Revert "Remove after_success"

This reverts commit 1a6a8858f79339e9bcb830cd913114f164937ce8.