fix issue #3

Here is my litte fix #3