Change import statement to support Flask version 0.10