Minor changes to Integration Test class