Patch/fix tests