Fix missing callback