unittest fix proposal