Unit tests correction.