test: fix after merge