Fix test relying on implementation details. Bug: 272403971 Test: treehugger Change-Id: I5483155eea84b1b14051bda7107d4caaf2687c93