fixed issue #1

fixed issue #1 :)