flask tests fix