fix unit tests
use custom entity label plugin