A couple small changes that I forgot on last push