Minor update to fix two comments