Fix #4

Fixed issue #4