Fixed unit test error.