updates for v2 fork