Update for flask-script 2.x