fix bug in some special case

Change-Id: I9af177c7a6593b0a9b68190dfbd5942f222485d4