Issue 5

fixed #5