This is a fix

fixes #2 

Here's a fix.