Fix after_success notes generation