fixed #3 issue

I fixed issue #3. Please merge this patch.