Fix unit test because of html change