tests: fix after protocol change