Add tests for the callback.