Minor edit to fix issue #125
Sorry my bad, fixes the wrong line
Merge pull request #126 from gabriel-kaam/master

Minor edit to fix issue #125