fix in some tests