unit tests correction