Updated test and comment