Revert "Added a TODO for integration test"

This reverts commit 89a859c5238ad6675b7687e8d79bb79ada9a06c9.