Add tests (one broken)