Fixed minor flask issue