Fixes for v2