moved callback but forgot to include in last commit