Update test and comment