fix tests that failed