Fix broken code.

Code that was missed from
986960dbdc82e968026315735213d509ec30ce2a

Change-Id: I95eb916387c91a9599006d7830cb2779455f003c