Fix getting init image layout