small fix for issue #2