Fix initial code
Merge pull request #1 from cielavenir/patch-1

Fix initial code