Fixed bug I made
Merge pull request #185 from EnacheB/master

Fixed bug I made