Fixed error on unit test.