Add a test for entitites