Fixed github issue #3
Minor change