Fix entities test case