Minor comments change

Didn't update comments after a minor design change