fixed entity test due to the last changes