Fix code example
Merge pull request #24 from aebm/patch-2

Fix code example