Test that the correct callback is used