fix #2

as suggest in the Issue #2