The original code was only taking the first category, and depending on how the terms are sorted then it may have chosen one without an image. If you wanted to always use the top level category then you’d need to loop through the list of terms and find the one with the parent of 0.
