fixes some unit tests