Minor bug fix.
Merge pull request #61 from vrege/master

Minor bug fix.