fix corresponding unit test