Small unittest fix.