Fixed issue in unit tests.