Fixes issue with model setting.

The image was updated even when there is no need for it. This fixes it.