Added a TODO for integration test