Add missing callbacks