Minor unittest fix.