Fix test warnings