A fix for #38

It's much simpler than everyone else thought. Sorry I couldn't commit this earlier, I've been at school!