corrections in unit tests