Apply changes to missed one line
Merge pull request #36 from yous/patch-1

Apply changes to missed one line