unit testing fix