fix tests to work