Fix #7

But it's a very very simple fix. Might want to do something more sophisticated.