fix test

Change-Id: I941c08b25e747eb15e84b12d3a7d7370d4c418f0