fix tests

Should fix tests.