Fixed flask tests