Cleanup

Small cleanup.