Minor code tweak.

Change-Id: I50b6f0f54026663561b55f48973eb91266a2cf67