Tidy a couple of minor mistakes