Bug fix

Mmm, bugs.