Small fix for issue #127