Fix incorrect Unit Test