Minor bug fix.

Change-Id: I4cae57bb5fb406b408c922a6727a70897d08f4e6