Fix test and comment