Fix mistral test_callback