Fixes #4

Fixed issue #4