Corrected Integration Test Setup