Add an integration test