refactor unit test because of the addition of the mainImageUrl model property