Unittest fix.