Add tests.
Fix.