Fix #7. I'm not really sure why.