Add missing callback params