Fix protocol tests