fix test and comment