fixed the unit test for the previous fix