Added an integration test