fixed bug, issue #2