Fixed some code issue unit tests.