Fixed issue #3 --amend #2