Added missing callback in test