Small fix to issue #16
Merge pull request #17 from tty02-fl/master

Small fix to issue #16