More test fixes <_<