Unittest fix(?)