Fix the test_protocol