Minor cleanup
Tiny cleanup