Updated Integration Test