small change in unit tests