Update callback declarations in tests.