fix tests so that they are passing