Made changes to Integration Tests