State script based on image

Hi Everybody,
In a production environment I have same devices with different images that are on customer site.
On these devices we are going to deploy a major update, so we designed a state script that is going to convert the configuration file of our utility to be compatible with the new utility version.

Some of the configuration that the new utility needs are missing, and these are different from customer to customer.

Is there a way to include a different state script based on the image selected?
image-customer-1 → state-script-customer1
image-customer-2 → state-script-customer2

Thanks!
Filippo