Fixed bugs (?) for v2