Fix for test_callback