small fix in flit class