Add tests
Fixes