Fixed issue #8

self explanatory