fix integration with ut-image