add missing onSuccess callback