Updated comment in line with review suggestion