Fix for the below commit

Change-Id: I04f4133ea3d72a8cbe20df91f5592460dd962beb