fix for issue #2

I want to fix #2