Update IntegrationTest.java