unittest fix