Fixed error in unit test.