Fix typo and update code for PR #1