Fixed up missing callback