More tests

Ah, right. I didn't think about that. Will fix.