tests: Small correction to make test case more stable

Change-Id: I720499212a3857df2e1467cfee8c7e1f4decdf78