[Tests] fix number of expected queries