Fix what I missed

Change-Id: If868c02c29a354653587cefc3cd0a817fa2cc176