Fix issue #3

Hmm... need some more time to get around Github. Hopefully I can link my PR with my issue somehow. :flushed: