Fix small bug

Change-Id: Iccfd0c5c1fa2166e4f34449a11d2490bc90a78b6