Minor fix to support Pillow