Tiny cleanup :)