correction v2 + tests