Fixed failing integration test