Re-add missing callback