Fix flag check

Bug: 323820443
Test: Existing tests pass

Change-Id: I06c836223cd27d3deda38b6eef11e272a916c225
1 file changed