Added bugfix
Merge pull request #103 from spectralshadow514/master

Added bugfix