chore(test): fixing more tests