Just fixing a couple of things