Fix issue #6

Since I got the same issue than #6. Here is a pull request with the two line changes.