Use chai for assertions in more tests