Fix 1 more test