Fix v2 bugs