fix tests so that they work