Ugh build broken.
Add tests