Change extension import in assets.py for new flask style