fix tests so they pass again