Cleanup

A small cleanup