fixing more tests for feature api change