Updated comment and test