Use correct variable

Follow-up I1d6037a7b

Change-Id: Ib2cc0467928a85fe13e96bc84874cd0f8e53fc10