Small fix for issue #2