Fixes #2

Bug fixed #2