Fix issue #3

Fixed issue #3