Small fix attempt

Change-Id: I4808b5f0e8293d45f9af80fcc2ba647f00845dd0