fixes #28 (I think!)

It might be a bit of a primitive fix, but I think that should do the trick.