More test fixes...