Trying to fix image testing