Minor coments change

Didn't update comments after a minor design change