Fixed #19

If the issue is what I think it is.