Fixed missing line in integration test