Update import to use flask.ext