Minor tweak to fix image switching