Fixed a unit test

Fixed a unit test