Minor cleanup

Really minor