Tiny cleanup. Fixes #1