Tweaks to fix image