small cleanup

Hey! I think I found a couple of details that can be cleaned up from your contribution.  Let me know what you think.