Fixed broken callback