fix for some tests