Quick test fix