Control Flow is incorrectly produced for code containing "with" statement

[datapaf]

Hello! Trying to produce control flow graph for the following code:

def get_fun(fun):
    with _get_serv(ret=None, commit=True) as cur:
        sql = 'SELECT s.id, s.jid, s.full_ret FROM salt_returns s JOIN (SELECT MAX(`jid`) as jid from salt_returns GROUP BY fun, id) max ON s.jid = max.jid WHERE s.fun = %s'
        cur.execute(sql, (fun,))
        data = cur.fetchall()
        ret = {}
        if data:
            for (minion, _, full_ret) in data:
                ret[minion] = full_ret
        return ret

I get the following graph:

I believe this graph does not represent the actual control flow of the code