Small bug fix.

Change-Id: I83db00d1bffcc662d81b144895b32c77ff3a341a