Fixed unit test bug.