Add tests integration