Fix tests after a small change in Wit API

A small change have been made in Wit API /entities.

I fixed the tests according to the change.