Fix integration test for ML resource