Misc. fixes (thanks @jayschwa)