Fix for the below commit

Change-Id: Ie96678cc97fd70ea462be672a0ceda154bae2b8c