Fix typo of the word "assign" in comments