fix tests

fix broken tests on master.