Fix ImageModel creation