improve withClass contains Word test
fixing test