Add: Integration test