Move comment to correct function