Fix callback in test