small code fix in unit testing