Fixes #46 for v2