Fix the import for pillow style