Small fix issue #1