Fix failing test in image collection