Unittest fix