Fix test

Change-Id: If2f19f77588b741a27187f5c43f8e79bbab99d66