Fix for issue #4

Yehhhp.