Fix incorrect comment

Comment had not been updated along with the code.