Added TODO in integration testing