More test fix