Fix old Flask Extension import error