Modified a unit test which was actually incorrect behavior in version 1.