fixes #3 issue

yes^ fixes #3