UPDATED fixed a unit test.