Fix 2 test warnings