partial fix of test