Added missing callback registration code