fix some test related to previous commit