Update test to match protocol