Adding integration test. Not working