Use entity_test module insted of adding custom entity.