Minor comments change

Didn't update comments after a minor design change :sad: