Add test regen. Fixes #2